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

最新推荐文章于 2017-11-08 23:33:30 发布

原创最新推荐文章于 2017-11-08 23:33:30 发布 · 1.2k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#矩阵快速幂 #二项式定理 #数论

ACM and novicer 同时被 3 个专栏收录

105 篇文章

订阅专栏

数学-组合数学

8 篇文章

订阅专栏

数学-矩阵

5 篇文章

订阅专栏

本文详细介绍了如何解决LightOJ1132问题，即计算从1到N的整数k次幂的累加和对2^32取模。通过应用二项式定理和构造变化矩阵的方法，实现了高效的计算过程，并提供了C++实现代码。此题旨在提升选手对数学和算法的理解与运用能力。

LightOJ 1132

题意：

$给出N (1 ≤ N ≤ 10^{15}) ， K (0 ≤ K ≤ 50)；$
$计算：(1^K + 2^K + 3^K + ... + N^K) \% 2^{32}.$

思路：

根据二项式定理，我们可以得到：
$(n+1)^k=C_k^0n^k+C_k^1n^{k-1}+C_k^2n^{k-2}+...+C_k^{k-1}n+1$

于是可以构造变化矩阵：

⎛ ⎝ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ C 0 k 00 . . . 01 C 1 k C 0 k - 1 0 . . . 00 C 2 k C 1 k - 1 C 0 k - 2 . . . 00 C 3 k C 2 k - 1 C 1 k - 2 . . . 00 . . . . . . . . . . . . . . . . . . 111 . . . 11 ⎞ ⎠ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ * ⎛ ⎝ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ n k n k - 1 . . . n 1 s u m ⎞ ⎠ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ = ⎛ ⎝ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ ⎜ (n + 1) k (n + 1) k - 1 . . . n + 1 1 s u m ⎞ ⎠ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟ ⎟

$\left( \begin{array}{ccc} C_k^0&C_k^1&C_k^2&C_k^3&...&1\\ 0&C_{k-1}^0&C_{k-1}^1&C_{k-1}^2&...&1\\ 0&0&C_{k-2}^0&C_{k-2}^1&...&1\\ ...&...&...&...&...&...\\ 0&0&0&0&...&1\\ 1&0&0&0&...&1\\ \end{array} \right) * \left( \begin{array}{ccc} n^k\\ n^{k-1}\\ ...\\ n\\ 1\\ sum\\ \end{array} \right)= \left( \begin{array}{ccc} (n+1)^k\\ (n+1)^{k-1}\\ ...\\ n+1\\ 1\\ sum\\ \end{array} \right)$
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;
}