1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
| #include<stdio.h> #include<cstring> #include<algorithm> #include<iostream> #define mem(x,y) memset(x,y,sizeof(x)) #define inf 0x3f3f3f3f using namespace std; #define siz 100 #define Mytype int int N; Mytype mod; struct matrix { Mytype a[siz][siz]; matrix operator*(const matrix &y)const { matrix res; mem(res.a,0); for(int i=0;i<N;i++) for(int j=0;j<N;j++) if(a[i][j]) for(int k=0;k<N;k++) res.a[i][k]+=a[i][j]*y.a[j][k]; return res; } matrix operator+(const matrix &y)const { matrix res; for(int i=0;i<N;i++) for(int j=0;j<N;j++) res.a[i][j]=a[i][j]+y.a[i][j]; return res; } }; matrix qmod(matrix a,long long k) { matrix res; mem(res.a,0); for(int i=0;i<N;i++) res.a[i][i]=1; while(k) { if(k&1) res=res*a; a=a*a; k>>=1; } return res; } int main() { int m,k; scanf("%d%d%d",&N,&m,&k); int t=k%m; matrix a,b; mem(a.a,0); for(int i=0;i<N;i++) a.a[i][i]=1; b=a; for(int i=1;i<=m;i++) { matrix c; mem(c.a,0); for(int j=0;j<N;j++) { int tmp; scanf("%d",&tmp); c.a[j][tmp-1]=1; } a=c*a; if(i==t)b=a; } a=qmod(a,k/m); a=b*a; for(int i=0;i<N;i++) for(int j=0;j<N;j++) if(a.a[i][j])printf("%d ",j+1); }
|