[六省联考2017]组合数问题
生活随笔
收集整理的這篇文章主要介紹了
[六省联考2017]组合数问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目大意:求這個式子:
∑i=0∞Cik+rnk(modp)這道題就是個數學題,做法其實就是優化的暴力——楊輝三角矩陣加速。
可以加速的原理,其實就是楊輝三角是一個一維遞推,并且可以將遞推描述為:復制矩陣到一個新矩陣,然后矩陣右移一格,加到新矩陣中。
對于這個題來說就可以用k?1的初始矩陣來乘以一個k?k的遞推矩陣的n?k次方。
然后矩陣快速冪就好了。
答案是矩陣的第r個元素。
代碼: #include<bits/stdc++.h> #define ll long long using namespace std; ll p; struct matrix{int r,c;ll m[52][52];void init(){memset(m,0,sizeof(m));}inline matrix operator * (const matrix& B) const {matrix C;C.r=r;C.c=B.c;C.init();for(int i=0;i<r;i++){for(int j=0;j<B.c;j++){for(int k=0;k<c;k++){C.m[i][j]=(C.m[i][j]+m[i][k]*B.m[k][j])%p;}}}return C;} }; ll n,k,r; matrix ans,B; matrix ksm(matrix A,ll b){while(b){if(b&1){ans=A*ans;}A=A*A;b>>=1;}return ans; } int main(){scanf("%lld %lld %lld %lld",&n,&p,&k,&r);ans.init();ans.r=k;ans.c=1;ans.m[0][0]=1;B.init();B.r=k;B.c=k;for(int i=0;i<k;i++){B.m[i][i]=1;B.m[i][(i+1)%k]++;}ans=ksm(B,n*k);printf("%lld",ans.m[r][0]);return 0; }
轉載于:https://www.cnblogs.com/stone41123/p/7581265.html
總結
以上是生活随笔為你收集整理的[六省联考2017]组合数问题的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: patch.exe是什么
- 下一篇: input 打开文件夹事件委托