题目描述
有n个人依次排队打饭,有m种饭菜可以选择,每个人可能选择其中一种,如果相邻排队的人打的菜一样,那么就会影响彼此吃饭的心情,求这个队伍中有人被影响心情的状态数,对100003取余。
样例解释
000 001
011 100
110 111
6种
输入
一行整数依次给出m,n
输出
相应的状态数
输入样例复制
2 3
输出样例复制
6
Data Constraint
30%:1<=n<=20,1<=m<=2
100%: 1<=m<=108,1<=n<=1012
题解:
考虑每个相邻人之间的菜不同的个数
及,第一个人随便选,之后每个人只有 m - 1 种选择可能
及, ( m - 1 ) n-1 * m
则最后答案为:
m * ( m n-1 - ( m - 1 ) n-1 )
Code
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#define LL long long
#define ull unsigned LL
#define db double
#define ldb long db
#define re register
using namespace std;
const LL N=1e6+100,M=2020,mod=100003;
LL n,m,sb;
LL kpow ( LL n,LL m )
{
LL ans = 1;
while ( m )
{
if ( m&1 ) ans=ans*n%mod;
m>>=1;
n=( n*n )%mod;
}
return ans;
}
int main ( )
{
scanf ( "%lld %lld",&m,&n );
LL sb=m*( kpow ( m,n-1 )+mod-kpow ( m-1,n-1 ) )%mod;
printf ( "%lld",sb );
return 0;
}