涂色问题

涂色问题
有N个块连成环，K种色，要求邻块异色，问涂色方案数

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,k;cin>>n>>k;//N块，K色
    if(k==1){//如果只有一种色
        if(n==1)cout<<"1"<<endl;//如果只有一块，就一种涂法
        else cout<<"0"<<endl;//超过一块，零种
        return 0;
    }
    if(k==2){//如果有两种色
        if(n%2==1)cout<<"0"<<endl;//有奇数个块成环就零方法
        else cout<<"2"<<endl;//有偶数个块成环就两种
    }
    if(n==1){cout<<k<<endl;return 0;}//只有一个块，有K种色
    if(n==2){cout<<k*(k-1)<<endl;return 0;}//有两个块，第一块有K种，第二块有K-1种，乘起来
    int dp[100][2];//开DP数组，当然也是可以只开两个变量的，滚动使用
    dp[2][0]=0;dp[2][1]=k*(k-1);//第二块跟第一块同色的方案为0，异色的方案是k*(k-1)
    for(int i=3;i<=n;i++){//从3到N扫一次，每次更新DP[i][0]与DP[i][1]
        dp[i][0]=dp[i-1][1];//第I块与第1块同色的方案数=第I-1块与第1块异色方案数
        dp[i][1]=dp[i-1][1]*(k-2)+dp[i-1][0]*(k-1);//第I块与第1块异色的方案数=第I-1块与第1块异色方案数*（K-2）+第I-1块与第1块同色方案数*（K-1）
    }
    cout<<dp[n][1]<<endl;
    return 0;
}