CodeForces 980 D Perfect Groups

Perfect Groups

题意：对于每一个连续的区间，区间内的数至少分成几个组，使得每个组内的数任意2个相乘是一个平方数。 输出的话是每个组数的个数。

题解：对于每一个数都将将他处理成不含平方数因子的数， 这样只有本身乘本身才会是一个平方数。

 　　  然后对这个数组处理出这个数前面的位置， 然后 n^2跑出答案的时候， 开一个cnt记录数字出现的个数记录，每次扫到一个数的时候，只有这个数上次出现的位置在起点之前 才执行cnt++;

　　  注意的是 如果 该数是0那么也不执行cnt++; 因为0在这个题目中也算完全平数。

代码：

#include<bits/stdc++.h>
using namespace std;
#define Fopen freopen("_in.txt","r",stdin); freopen("_out.txt","w",stdout);
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
const int N = 1e4+10;
int vis[N], prim[N], cnt[N], pre[N], A[N];
int tot = 0, ans = 0;
map<int,int> mp;
void solve(int u){
    for(int i = 0; i < tot; i++){
        int t = prim[i] * prim[i];
        if(abs(A[u]) < t) return;
        while(A[u]%t == 0){
            A[u] /= t;
        }
    }
}
int main(){
    for(int i = 2; i < N; i++){
        if(!vis[i]){
            prim[tot++] = i;
            for(int j = i; j < N; j += i)
                vis[j] = 1;
        }
    }
    int n;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
        scanf("%d", &A[i]), solve(i);
    for(int i=1; i<=n; i++) {
        pre[i] = mp[A[i]];
        mp[A[i]] = i;
    }
    for(int i=1; i<=n; i++) {
        int cur=0;
        for(int j=i; j<=n; j++) {
            if(A[j] && pre[j] < i && A[j] != 0) cur++;
            cnt[max(1,cur)]++;
        }
    }
    for(int i = 1; i <= n; i++)
        printf("%d%c", cnt[i], " \n"[i==n]);
    return 0;
}