P2926 [USACO08DEC]Patting Heads S(简单数学+装桶)

P2926 [USACO08DEC]Patting Heads Shttps://www.luogu.com.cn/problem/P2926

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <cstring>
#include <set>
#include <cmath>
#include <map>
#include <cstdlib>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
const int MN = 65005;
const int MAXN = 2000010;
const int INF = 0x3f3f3f3f;
#define IOS ios::sync_with_stdio(false)


int n;
int a[MAXN];
int b[MAXN];


int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		scanf("%d", a + i);
		b[a[i]]++;
	}
	for (int i = 1; i <= n; i++) {
		int ans = 0;
		b[a[i]]--;
		for (int j = 1; j <= sqrt(a[i]); j++) {
			if (a[i] % j == 0) {
				ans += b[j];
				if (a[i] != j * j) {
					ans += b[a[i] / j];
				}
			}
		}
		printf("%d\n", ans);
		b[a[i]]++;
	}
	return 0;
}