AT_abc439_c [ABC439C] 2026

思路

exactly one pair

注意到是“恰好一对”。
枚举 $x,y$，将所有合法的 $x^2+y^2$ 加入一个数组内。
这样的数最多有 $n$ 个。
考虑对这个数组排序。
在排序后的数组内统计答案，若一个数既不与其前一个相等，也不与其后一个相等，因为数组是有序的，则它对应的 $x,y$ 一定只有一对。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include<bits/stdc++.h>
#define fir first
#define sec second 
#define int long long
#define pii pair<int,int>
#define fep(i,s,e) for(int i=s;i<e;i++)
#define pef(i,s,e) for(int i=s;i>e;i--)
#define rep(i,s,e) for(int i=s;i<=e;i++)
#define per(i,s,e) for(int i=s;i>=e;i--)
namespace FastIO{
	template<typename T>inline void read(T &x){
		x=0;int f=1;char c=getchar();
		for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
		for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+(c^48);x*=f;
	}
	template<typename T,typename...Args>
    inline void read(T &x,Args&...args){
        read(x);
        read(args...);
    }
	template<typename T>void print(T x){
		if(x<0)x=-x,putchar('-');
		if(x>9)print(x/10);
		putchar((x%10)^48);
	}
}
using namespace std;
using namespace FastIO;
int n,cnt;
vector<int>v;
signed main(){
	read(n);
	rep(i,1,n){
		if(i*i>n)break;
		rep(j,i+1,n){
			if(i*i+j*j>n)break;
			v.emplace_back(i*i+j*j);
		}
	}
	sort(v.begin(),v.end());
	fep(i,0,v.size()){
		if(v[i]!=v[i-1]&&v[i]!=v[i+1]){
			cnt++;
		}
	}
	print(cnt);
	puts("");
	fep(i,0,v.size()){
		if(v[i]!=v[i-1]&&v[i]!=v[i+1]){
			print(v[i]);
			putchar(' ');
		}
	}
}