AT T4 求hack
查看原帖
AT T4 求hack
902351
Little_x_starTYJ楼主2024/2/24 22:00
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
int k[2000];
int id;
int a[200010];
int c[200010];
int d[200010];
ll ans;
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	for (int i = 0; i * i <= 200010; i++) {
		k[++id] = i * i;
	}
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		c[a[i]]++;
	}
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= id; j++) {
			if (a[i] == 0) {
				ans += n - 1;
				break;
			}
			if (k[j] >= a[i] || k[j] == 0) {
				if (k[j] % a[i] == 0 && c[k[j] / a[i]]) {
					if (k[j] / a[i] == a[i]) {
						if (c[k[j] / a[i]] > 1)
							ans += c[k[j] / a[i]] - 1;
					}
						
					else
						ans += c[k[j] / a[i]];
				}
			}
		}
	}
	cout << ans / 2;
	return 0;
}
2024/2/24 22:00
加载中...