【数学】JZOJ_4248 n染色

题意

有一个每条边都不相等的$N$边形，有$M$种颜色。我们要给每条边涂上颜色使得相邻的两条边都不相等，求总方案数。

思路

设$f_i$为$i$条边时的答案，那么：$f_i=m^i-f_{i-1}$，代表用总的方案数减去不合法的方案数。

从$f_{i-1}$加一条边有$m^i$种方法。不合法的方案：我们把新加的边和它相邻的一条边看成颜色相同的同一条边，这是不合法的，这种方案数有$f_{i-1}$种。
然后可以发现：
$f_i=m^i-(m^{i-1}-(m^{i-2}-f_{i-3}($…一直到1$))$，然后一直到$1$，我们就可以发现其中的规律：
$$f_n=(m-1{)}^n+(m-1)\ast (-1{)}^n$$

代码

#include<cstdio>

const int P = 1e9 + 7;
long long n;
int m;

int power(int a, long long b) {
	a %= P;
	int result = 1;
	for (; b; b >>= 1) {
		if (b & 1) result = (long long)result * a % P;
		a = (long long)a * a % P;
	}
	return result;
}

int main() {
	scanf("%lld %d", &n, &m);
	printf("%d", power(m - 1, n) + (m - 1) * power(-1, n) % P);
}