Uva1635-无关的元素

题目链接

Uva1635-无关的元素

---

题意

对于给定的n个数a1,a2,…,an,依次求出相邻两之和，将得到一个新数列，重复上述操作姜变成一个数，问这个数除以m的余数与哪些数无关？例如n = 3,m = 2时，第一次求和得到a1+a2,a2+a3,再求和得到a1+2a2+a3,他除以2的余数和a2无关。1<=n<=10^5,2<=m<=10^9

---

题解

最后的合式是a1,a2,…,an的线性组合，设ai的系数为f(i),则合式除以m的余数与ai无光。当且仅当f(i)是i的倍数。
最后合式的ai的系数是Ci-1/n-1,这样问题就变成了C0/n-1,C1/n-1，…,Cn-1/n-1中有哪些是m的倍数，我们只需要关心哪些ai是m的倍数，只需要依次计算m的唯一分解式中各个素因子再ai中的指数即可完成判断。

---

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
vector<int> primes,ans;
int e[100],ec[100];
void solve(int n)
{
    memset(e,0,sizeof(e));
    memset(ec,0,sizeof(ec));
    ans.clear();
    primes.clear();
    int num = 0;
    for(int i=2;i*i<=n;i++)
    {
        if(n % i == 0)
        {
            primes.push_back(i);
            while(n % i == 0)
            {
                n /= i;
                e[num]++;
            }
            num++;
        }
    }
    if(n > 1) {
        primes.push_back(n);
        e[num]++;
    }
}
void add_int(int x,int d)
{
    for(int i=0;i<primes.size();i++)
    {
        while(x % primes[i] == 0) {
            x /= primes[i];
            ec[i] += d;
        }
    }
}
bool check()
{
    for(int i=0;i<primes.size();i++)
        if(ec[i]<e[i]) return false;
    return true;
}
int main()
{
    int q,n;
    while(~scanf("%d%d",&q,&n))
    {
        q--;
        solve(n);
        if(check()) ans.push_back(1);
        for(int i=1;i<=q;i++)
        {
            add_int(q-i+1,1);add_int(i,-1);
            if(check()) ans.push_back(i+1);
        }
        printf("%d\n",ans.size());
        for(int i=0;i<ans.size();i++) printf("%s%d",i==0?"":" ",ans[i]);
        printf("\n");
    }
    return 0;
}