JZ Day13 T2 n染色

题目大意：

有一个图形有$n$条边，某位$dalao$有$m$支不同颜色的笔，要给$n$条边都涂上颜色并且要求每两条相邻的边颜色不相同，求方案数

---

解题思路：

比赛的时候暴力打表找规律
发现当$m$相同的时候：
$f_i=f_{i-1}\times (m-2)+f_{i-2}\times (m-1)$
然后推了一推$f_n$与$f_1,f_2$的关系
没推出来

然后下面就是证明
首先我们在$f_1=m,f_0=0$的情况下证明$f_n+f_{n+1}=(m-1{)}^n\times m$
我们有：$f_{n+1}=f_n\times (m-2)+f_{n-1}\times (m-1)$
所以原式$=f_n+f_n\times (m-2)+f_{n-1}\times (m-1)$
合并同类项：原式$=f_n\times (m-1)+f_{n-1}\times (m-1)$
接着再把$f_n=f_{n-1}\times (m-2)+f_{n-2}\times (m-1)$带入
原式$=[f_{n-1}\times (m-2)+f_{n-2}\times (m-1)]\times (m-1)+f_{n-1}\times (m-1)$
化简得$f_{n-1}\times (m-2)(m-1)+f_{n-2}\times (m-1{)}^2+f_{n-1}\times (m-1)$
合并同类项：原式$=f_{n-1}\times (m-1{)}^2+f_{n-2}\times (m-1{)}^2$
……
以此类推，原式$=f_1(m-1{)}^n+f_0(m-1{)}^n=(m-1{)}^{n}m$

$\therefore f_n+f_{n+1}=(m-1{)}^{n}m$
但是我们要求的是$f_n$
所以我们想办法把把$f_n$或$f_{n+1}$消掉
我们尝试消去$f_n$，让两边同时加上或减去某些数，原等式左边为：
$f_n+f_{n+1}-(f_n+f_{n-1})+(f_{n-1}+f_{n-2})-(f_{n-2}+f_{n-3})+...\pm (f_1+f_0)$
$\because f_n+f_{n+1}=(m-1{)}^{n}m$
$\therefore$等式右边为：
$(m-1{)}^{n}m-(m-1{)}^{n-1}m+(m-1{)}^{n-2}m-(m-1{)}^{n-3}+...\pm m$
我们发现这是一个等比数列，公比为$-(m-1)=1-m$,
但是我们不确定最后的符号
所以我们要判断$n$的奇偶性
当$n$为偶数时最后的符号为$"-"$
否则为$"+"$
然后我们暂且不考虑最后一项$\pm m$
也就是说我们共有$n-1$项
然后我们用等比数列求和公式($3\le m\le 50$所以$1-m$即公比不可能是一所以我们只需要考虑公比不为1的情况)：
设$Q=1-m$(公比)
$S_n=\frac{f_1\times (1-Q^{n-1})}{1-Q}$
我们试着带入化简：
$\because f_1=m,Q=1-Q$
$\therefore$原式$=f_n=\frac{m\times (1-Q^{n-1})}{1-(1-m)}=\frac{m\times (1-Q^{n-1})}{m}=1-Q^{n-1}$
再判断一下$n$的奇偶性加上最后一项就好了

---

$Acceptedcode:$

#include<cstdio>

using namespace std;

const int ymw = 1e9 + 7;

int m;
long long n;

long long ksm(long long x, long long m) {
	long long ans = 1; x %= ymw;
	while (m) {
		if (m & 1) (ans *= x) %= ymw;
		(x *= x) %= ymw;
		m >>= 1;
	}
	return ans;
}

int main() {
	//freopen("color.in", "r", stdin);
	//freopen("color.out", "w", stdout);
	scanf("%lld %d", &n, &m);
	printf("%d", (ksm(m-1, n) + (m-1) * ((n&1) ? -1 : 1)) % ymw);
}