20210809D

题目描述

有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<=10⁸,1<=n<=10¹²

题解：

考虑每个相邻人之间的菜不同的个数

及，第一个人随便选，之后每个人只有 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;
}