CF 71C. Round Table Knights

题目链接

很多小的细节都没想到。。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
int p[100001];
int o[100001];
int main()
{
    int n,i,flag = 1,j;
    scanf("%d",&n);
    for(i = 0;i < n;i ++)
    {
        scanf("%d",&p[i]);
        if(p[i] == 0)
        flag = 0;
    }
    if(flag)
    {
        printf("YES
");
        return 0;
    }
    if(n%2 == 0&&n > 4)
    {
        memset(o,0,sizeof(o));
        for(i = 0;i < n;i ++)
        {
            o[i%2] += p[i];
        }
        if(o[0] == n/2||o[1] == n/2)
        {
            printf("YES
");
            return 0;
        }
    }
    for(i = 3;i*i <= n;i ++)
    {
        if(n%i == 0)
        {
            int mod;
            memset(o,0,sizeof(o));
            mod = n/i;
            for(j = 0;j < n;j ++)
            {
                o[j%mod] += p[j];
            }
            for(j = 0;j < mod;j ++)
            {
                if(o[j] == i)
                {
                    printf("YES
");
                    return 0;
                }
            }
            memset(o,0,sizeof(o));
            mod = i;
            for(j = 0;j < n;j ++)
            {
                o[j%mod] += p[j];
            }
            for(j = 0;j < mod;j ++)
            {
                if(o[j] == n/i)
                {
                    printf("YES
");
                    return 0;
                }
            }
        }
    }
    printf("NO
");
    return 0;
}