HDU 5908 Abelian Period 模拟乱搞

题目：http://acm.hdu.edu.cn/showproblem.php?pid=5908

题意：

问题描述

设SS是一个数字串，定义函数occ(S,x)occ(S,x)表示SS中数字xx的出现次数。

例如：S=(1,2,2,1,3),occ(S,1)=2,occ(S,2)=2,occ(S,3)=1S=(1,2,2,1,3),occ(S,1)=2,occ(S,2)=2,occ(S,3)=1。

如果对于任意的ii，都有occ(u,i)=occ(w,i)occ(u,i)=occ(w,i)，那么我们认为数字串uu和ww匹配。

例如：(1,2,2,1,3)\approx(1,3,2,1,2)(1,2,2,1,3)≈(1,3,2,1,2)。

对于一个数字串SS和一个正整数kk，如果SS可以分成若干个长度为kk的连续子串，且这些子串两两匹配，那么我们称kk是串SS的一个完全阿贝尔周期。

给定一个数字串SS，请找出它所有的完全阿贝尔周期。

输入描述

输入的第一行包含一个正整数T(1\leq T\leq10)T(1≤T≤10)，表示测试数据的组数。

对于每组数据，第一行包含一个正整数n(n\leq 100000)n(n≤100000)，表示数字串的长度。

第二行包含nn个正整数S_1,S_2,S_3,...,S_n(1\leq S_i\leq n)S1,S2,S3,...,Sn(1≤Si≤n)，表示这个数字串。

输出描述

对于每组数据，输出一行若干个整数，从小到大输出所有合法的kk。

思路：枚举所有的k去验证，因为k必须是n的约数，所以需要去验证的k并不多。统计所有不同的数字，把数组划分成n / k段，统计第一段每个数字出现的次数，之后比较每段数字出现的次数和第一段出现的次数，不同的说明k不可行

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

typedef long long ll;
const int N = 100010;
int a[N], b[N], brr[N], num[N], num1[N];
int n;
void reset(int k, int num[])
{
    for(int i = 0; i < k; i++)
        num[b[i]] = 0;
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        for(int i = 0; i < n; i++) scanf("%d", &a[i]), b[i] = a[i];
        sort(b, b+n);
        int len = unique(b, b+n) - b;
        int res[N], cnt = 0;
        for(int i = 1; i <= n; i++)
            if(n % i == 0)
            {
                int flag = 1;
                for(int j = 1; j <= n/i; j++)
                {
                    if(j == 1)
                        for(int k = (j-1)*i; k < i*j; k++)
                            num[a[k]]++;
                    else
                    {
                        int tot = 0;
                        for(int k = (j-1)*i; k < i*j; k++)
                            num1[a[k]]++, brr[tot++] = a[k];
                        sort(brr, brr + tot);
                        tot = unique(brr, brr + tot) - brr;
                        for(int k = 0; k < tot; k++)
                        {
                            if(b[k] != brr[k])
                            {
                                flag = 0; break;
                            }
                            if(num1[b[k]] != num[b[k]])
                            {
                                flag = 0; break;
                            }
                        }
                        reset(len, num1);
                    }
                    if(!flag) break;
                }
                if(flag) res[cnt++] = i;
                reset(len, num);
                reset(len, num1);
            }
        for(int i = 0; i < cnt; i++) printf("%d%c", res[i], i == cnt-1 ? '\n' : ' ');
    }
    return 0;
}