Codeforces Round #480 (Div. 2) D - Perfect Groups 数论+暴力

题意：

首先对于一个序列X，设一个值为ans1，ans1表示把这个序列划分成的最小组数，使得每一组中任意两个数相乘是平方数

对与本题：给定原序列A，对于A的任意一个连续子序列X，都有一个ans1，问A序列所有子序列的ans1都有多少个；

输出ans[i]，表示ans1为i的子序列的个数；

思路：

首先对于每个数，如果它有 平方数因子，那可以直接除掉，因为不影响，也为了方便后面计算ans1；O(n*max(a[i])）；同时对这个数hash成1~5000的一个数；（开始我用map直接标记超时了）；

n才5000，可以直接暴力找每个连续子序列，同时计算其ans1；

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5000 + 7;

int n, a[maxn], ans[maxn], b[maxn], id;
map<int, int> mp;
int t0 = 0, t1 = 0, t2 = 0;
int cnt[maxn];

void work(int id) {
    for(int i = 2; ; ++i) {
        int t = i*i;
        if(a[id] >= 0 && t > a[id]) return;
        if(a[id] < 0 && t > -1*a[id]) return;
        if(a[id]%t == 0) {
            while(a[id]%t == 0) {
                a[id] /= t;
            }
        }
    }
}

void init() {
    scanf("%d", &n);
    id = 1;
    for(int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
        work(i);
        int t;
        if(t = mp[a[i]]) b[i] = t;
        else b[i] = mp[a[i]] = id++;
    }
    memset(ans, 0, sizeof ans);
}

int main() {
    init();
    for(int i = 1; i <= n; ++i) {
        for(int k = 1; k < id; ++k) cnt[k] = 0;
        int sum = 0;
        for(int j = i; j <= n; ++j) {
            if(a[j] && !cnt[b[j]]) {
                sum++;
            }
            cnt[b[j]]++;
            ans[max(1, sum)]++;
        }
    }
    for(int i = 1; i <= n; ++i) {
        printf("%d%c", ans[i], (i == n ? '\n' : ' '));
    }
    return 0;
}