UVA-1635 数学

UVA-1635

题意：

给定n个数a1,a2,a3.....an,依次求出相邻的两个数的和，最后成为一个数，问这个数模m的值与那些最初的数无关

例：a1,a2,a3, m=2 => a1+a2,a2+a3 => a1+2a2+a3 ,显然第二项无关。

代码：

//可以看出最后得到的式子符合杨辉三角，所以可以递推出所有的项的系数(c(i+1)=c(i)*(n-i)/(i+1))，
//但是数太大不能直接计算,可以将他们分解成 a1^p1*a2^p2*... 的形式，这样就只需要判断因子的指数的大小即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=;
int c[maxn+];
int ans[maxn+],prime[maxn+],re[maxn+],tot;
void init(int m,int p){
    tot=;
    memset(re,,sizeof(re));
    for(int i=;i<=sqrt(m+0.5);i++){
        if(m%i==){
            prime[++tot]=i;
            while(m%i==){
                m/=i;
                re[tot]+=p;
            }
        }
    }
    if(m>){
        prime[++tot]=m;
        re[tot]+=p;
    }
}
int solve(int m,int p){
    for(int i=;i<=sqrt(m+0.5);i++){
        if(m%i==){
            while(m%i==){
                m/=i;
                c[i]+=p;
            }
        }
    }
    if(m>) c[m]+=p;
    for(int i=;i<=tot;i++){
        if(prime[i]>maxn) return ;
        else if(c[prime[i]]<re[i])
            return ;
    }
    return ;
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)==){
        init(m,);
        int cnt=;
        memset(c,,sizeof(c));
        n--;
        for(int i=;i<n;i++){
            solve(n-i,);
            int tmp=solve(i+,-);
            if(tmp) ans[++cnt]=i+;
        }
        printf("%d\n",cnt);
        for(int i=;i<=cnt;i++)
            printf("%d%c",ans[i],i==cnt?'\n':' ');
        if(cnt==) printf("\n");//坑！！！
    }
    return ;
}