【SC主题公园杯邀请赛】【cqbzoj1005】三个袋子 如何手残

三个袋子

时间限制: 1 Sec 内存限制: 64 MB

题目描述

平平在公园里游玩时捡到了很多小球，而且每个球都不一样。平平找遍了全身只发现了3个一模一样的袋子。他打算把这些小球都装进袋子里（袋子可以为空）。他想知道他总共有多少种放法。 将N个不同的球放到3个相同的袋子里，求放球的方案总数M。 结果可能很大，我们仅要求输出M mod K的结果。 现在，平平已经统计出了N<=10的所有情况。见下表：

N　　　　1　　　　２　　　　3　　　　4　　　　5　　　　6　　　　7　　　　8　　　　9　　　　10

M　　　　1　　　　2　　　　5　　　　14　　　　41　　　122　　　　365　　　1094　　3281　　　9842

输入

两个整数N,K，N表示球的个数。

输出

输出仅包括一行，一个整数M mod K 。

样例输入

11 10000

样例输出

9525

提示

数据规模 对于 40%数据,10<=N<=10,000 对于100%数据,10<=N<=1,000,000,000 对于 100%数据，K<=100,000

其实这道题有更简单的方法（数学差系列）
点这里看szp的题解

手残报告

找规律得f[i]=3*f[i-1]-1
明显矩阵加速

下面贴代码

#include<cstdio>
#include<cstring>
typedef long long LL;
LL mat_mod;
const int MAT_SIZE=4;
struct Matrix{
    int x,y;
    LL mat[MAT_SIZE][MAT_SIZE];
    Matrix(){x=y=0;memset(mat,0,sizeof mat);}
    void singlelize(){
        if(x!=y)puts("Error");
        memset(mat,0,sizeof mat);
        for(int i=1;i<=x;i++)
            mat[i][i]=1;
    }
};
Matrix operator*(const Matrix&a,const Matrix&b){
    if(a.y!=b.x)return *((Matrix*)0);
    Matrix ret;
    for(int i=1;i<=a.x;i++)
        for(int j=1;j<=b.y;j++)
            for(int k=1;k<=a.y;k++){
                ret.mat[i][j]+=(a.mat[i][k]*b.mat[k][j])%mat_mod;
                ret.mat[i][j]%=mat_mod;
            }
    ret.x=a.x;
    ret.y=b.y;
    return ret;
}
Matrix tmp;
Matrix adv_pow(Matrix a,LL b,LL c){
    mat_mod=c;
    tmp=a;
    tmp.singlelize();
    while(b){
        //printf("mtp");
        if(b&1)tmp=tmp*a;
        a=a*a;
        b>>=1;
    }
    return tmp;
}
int main(){
    LL n,k;
    scanf("%lld%lld",&n,&k);
    Matrix p;
    p.mat[1][1]=1;
    p.mat[1][2]=1;
    p.mat[2][2]=3;
    p.x=p.y=2;
    Matrix vec;
    vec.mat[1][1]=-1;
    vec.mat[1][2]=1;
    vec.x=1;
    vec.y=2;
    Matrix mul=adv_pow(p,n-1,k);
    vec=vec*mul;
    LL ans=vec.mat[1][2];
    if(ans<0)ans+=mat_mod;
    printf("%lld\n",ans);
}