码迷,mamicode.com
首页 > 其他好文 > 详细

LightOJ 1132 Summing up Powers(矩阵快速幂+二项式定理)

时间:2015-08-17 01:14:22      阅读:185      评论:0      收藏:0      [点我收藏+]

标签:矩阵快速幂   二项式定理   数论   

LightOJ 1132

题意:

N(1N1015)K(0K50)
(1K+2K+3K+...+NK)%232.

思路:

根据二项式定理,我们可以得到:
(n+1)k=C0knk+C1knk?1+C2knk?2+...+Ck?1kn+1

于是可以构造变化矩阵:

???????????C0k00...01C1kC0k?10...00C2kC1k?1C0k?2...00C3kC2k?1C1k?2...00..................111...11???????????????????????nknk?1...n1sum???????????=???????????(n+1)k(n+1)k?1...n+11sum???????????

sum 是 每一项N^K的累加和。

代码君:

/*
* @author FreeWifi_novicer
* language : C++/C
*/
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<string>
#include<map>
#include<set>
#include<vector>
#include<queue>

using namespace std;

#define clr( x , y ) memset(x,y,sizeof(x))
#define cls( x ) memset(x,0,sizeof(x))
#define mp make_pair
#define pb push_back
typedef long long lint;
typedef long long ll;
typedef long long LL;

const int maxn = 52;
const lint mod = (1LL << 32);
lint n,k;
int kase;
lint C[maxn][maxn];

struct Matrix{
    int n , m ;
    lint a[maxn][maxn];
    Matrix( int n , int m ){
        this->n = n ;
        this->m = m ;
        cls(a);
    }
    Matrix operator * ( const Matrix &tmp ){
        Matrix res( n , tmp.m );
        for( int i = 0 ; i  < n ; i++ )
            for( int j = 0 ; j < tmp.m ; j++ )
                for( int k = 0 ; k < m ; k++ )
                    res.a[i][j] = ( res.a[i][j] + ( a[i][k] * tmp.a[k][j] ) % mod ) % mod;
        return res;
    }
};

void Matrix_print( Matrix x ){
    for( int i = 0 ; i < x.n ; i++ ){
        for( int j = 0 ; j < x.m ; j++){
            cout << x.a[i][j] << ‘ ‘;
        }
        cout << endl;
    }
    cout << endl;
}

void init(){
    cls(C);
    C[0][0] = 1;
    for( int i = 1 ; i < maxn ; i++ ){
        C[i][0] = 1;
        for( int j = 1 ; j <= i ; j++ ){
            C[i][j] = C[i-1][j-1] + C[i-1][j];
            if( C[i][j] > mod ) C[i][j] -= mod;
        }
    }
}
Matrix fast_pow( Matrix x , lint n ){
    Matrix res( x.n , x.m );
    for( int i = 0 ; i < x.n ; i++ ) res.a[i][i] = 1;
    while( n ){
        if( n & 1 )
            res = res * x;
        x = x * x;
        n >>= 1;
    }
    return res;
}

void solve(){
    if(k == 0){
        printf( "Case %d: %lld\n" , kase++ , n % mod );
        return;
    }
    Matrix base( k + 2 , 1 );
    Matrix fun( k + 2 , k + 2 );
    fun.a[0][0] = 1;
    for( int i = 1 ; i < k + 2 ; i++ ) fun.a[0][i] = C[k][i - 1];
    for( int i = 1 ; i < k + 2 ; i++ ){
        for( int j = i ; j < k + 2 ; j++ ){
            fun.a[i][j] = C[k - i + 1][j - i];
        }
    }
    for( int i = 0 ; i < k + 2 ; i++ ) base.a[i][0] = 1;
    //Matrix_print( fun );
    //Matrix_print( base );
    fun = fast_pow( fun , n - 1 );
    base = fun * base ;
    printf( "Case %d: %lld\n" , kase++ , base.a[0][0] % mod );
}
int main(){
  //freopen("input.txt","r",stdin);
    int t ; cin >> t ; kase = 1;
    init();
    while( t-- ){
        cin >> n >> k ;
        solve();
    }
    return 0;
}



版权声明:博主表示授权一切转载:)

LightOJ 1132 Summing up Powers(矩阵快速幂+二项式定理)

标签:矩阵快速幂   二项式定理   数论   

原文地址:http://blog.csdn.net/qq_15714857/article/details/47708835

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!