首页 > 代码库 > BZOJ 1008: [HNOI2008]越狱【组合】

BZOJ 1008: [HNOI2008]越狱【组合】

很少有的思路秒解。题意可以描述成对长度为n的格子有m种染色方案,问存在相邻两个格子同色的方案数,正难则反易,考虑问题的背面任意两个相邻的格子都不同色,第一个格子可以涂任意一种颜色m种可能,剩下的n-1个格子每个不能和前面一个格子相同固有m-1种可能,总的方案数有m^n种,所以答案就是m^n-m(m-1)^n-1 由于答案取模及幂次较高,因此可以考虑小费马定理或快速幂优化   

由于数据太2,一开始用 int读成功贡献了3个WA TAT

 

#include<iostream>

#include<cstdio>

#include <math.h>

using namespace std;

long long ksm(long long k,long long a)

{

   if(k==0)return 1;

         if(k==1)return a;

    if(k % 2 == 0)return ksm(k/2,a)*ksm(k/2,a) % 100003;

   else return (ksm(k-1,a)*a) % 100003;

}

int main()

{

   long long m=0,n=0,ans;

   scanf("%lld%lld",&m,&n);

         ans=(ksm(n,m)-(m*ksm(n-1,m-1))% 100003 +100003)%100003;

         printf("%lld",ans);

   return 0;

}

BZOJ 1008: [HNOI2008]越狱【组合】