【DP+数学小技巧】三个袋子

1005: 三个袋子

时间限制: 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

一句话也不用说，一切都在代码中。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;

#define MAXN
#define MAXM
#define INF 0x3f3f3f3f
typedef long long int LL;

int N,K;

LL KSM(LL x,int p,int mod)
{
    LL rn=1;
    while(p)
    {
        if(p&1)
            rn=(rn*x)%mod;

        x=(x*x)%mod;
        p>>=1;
    }

    return rn;
}

int main()
{
    scanf("%d%d",&N,&K);

    /*
    其实这道题中最后一句话才是关键
    经过前10组数据的推导，很容易就可以发现
    f[N]=3^(N-1)*f[1]-(3^(i-2)+3^(i-3)+...+3^0);
        =(3^(N-1)+1)/2;

    但要对于K取模，如果求2的逆元，当K为偶数时无解
    这时候就要用到(a/b)%c=(a%(b*c))/b;

    具体证明：
    设a=i*b*c+j*b+k;
    则原式转换为:
        ((i*b*c+j*b+k)/b)%c=(i*b+j)%c=j;
      和((i*b*c+j*b+k)%(b*c))/b=(j*b+k)/b=j;
    左右相等，证毕。
    */
    printf("%d\n",(int)(KSM(3,N-1,K*2)+1)/2);
}