COCI2012/2013 Contest#4 E Solution

Description

可怕的流感在AC之城爆发了！AC之城有M个居民，每个人有一个ID编号，从0到M-1。由于病毒十分强大，每个人可能被多次感染。在流感爆发第一天，病毒携带者被我们称作“病原体”，他们的编号已知，病毒要靠他们传播。
在接下来的每一天，如果有一个前一天感染病毒的、编号为a的病人，以及一个编号为b的“病原体”，那么编号为p的居民将在今天被感染，其中p=(a*b)%M。另外，a和b不一定要严格不同。
举一个例子，假设AC之城有101个居民，其中5号和50号为“病原体”。在第一天，感染病毒的人为5号和50号。第二天，25号、48号(250%101)和76号(2500%101)将被感染。第三天，77号将成为在这天被感染的人中的一个((48*50)%101=77)。
现在，我们想要知道，将在第K天被感染？

Solution

出题人语文成绩与想象力成正比，与语文水平不成正比。。
愣是没看懂题面反而沉醉在情境里了，满脑子都在打瘟疫公司。。
这题好神的==
首先答案等于

$$\sum_{i=0}^{m-1}~\sum_{j=0}^{p_{s-1}} A[i]*B[j]~mod~m=\sum_{i=0}^{m-1}A[i]~\sum_{j=0}^{m}\sum_{j=k}^{p_{s-2}} A[j]*B[k]~mod~m=\sum_{i=0}^{m-1}~\sum_{j=0}^{p_{s-1}}~\sum_{k=0}^{p_{s-2}} A[i]*A[j]*B[k]~mod~m=...$$ $$=\sum_{i_0=0}^{m-1}\sum_{i_1=0}^{m-1}...\sum_{i_s=0}^{m-1}~~\prod_{j=0}^sA[i_j]~mod~m$$
于是可以用快速幂搞咯

Code

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
using namespace std;
typedef long long ll;
const int M=1505;
int f[M],A[M],n,m,w[M],ans[M];ll k;
inline void gao(){
    for(int i=0;i<m;++i)
        for(int j=0;j<m;++j)
            if(f[i]&&ans[j])w[i*j%m]=1;
    for(int i=0;i<m;++i)
        ans[i]=w[i],w[i]=0;
}
inline void Pow(){
    for(int i=0;i<m;++i)
        for(int j=0;j<m;++j)
            if(f[i]&&f[j])w[i*j%m]=1;
    for(int i=0;i<m;++i)
        f[i]=w[i],w[i]=0;
}
inline void solve(){
    for(int i=0;(1LL<<i)<=k;++i,Pow())
        if(k&1LL<<i)gao();
}
int main(){
    cin>>k>>m>>n;--k;
    for(int i=1;i<=n;++i)
        scanf("%d",&A[i]),f[A[i]]=ans[A[i]]=1;
    solve();
    for(int i=0;i<m;++i)
        if(ans[i])printf("%d ",i);
    puts("");
    return 0;
}