Wrong Answer on test 233 (Easy Version)------dp cf1261D1

     题目地址：https://codeforces.com/contest/1261/problem/D1

新手第一次~~~~

思路：
答案在顺时针旋转后，原来的正确数和旋转后的正确数的关系有两类
一、当这一题的答案和下一题的答案相同时，无论这一题选择任何一个选项对旋转后的结果没有影响。
二、当这一题的答案和下一题的答案不同时，有三种情况：
①这一题正确下一题不正确。
②这一题错误下一题正确。
③这一题和下一题都不正确。
由于第一类不会对旋转后的结果产生影响，所以我们只需考虑第二类，然后乘以第一类的种数。
dp:
横坐标表示：第几个这一题和下一题答案不同的数
纵坐标表示：旋转后的答案比没有旋转的答案多的正确数
转移方程为 dp[i][j]=dp[i-1][j-1]+dp[i-1][j+1]+dp[i-1][j]*(k-2)
代码如下：

#include<iostream>
using namespace std;
const int mo=998244353;
long long dp[2005][4005];
int ans[2000];
int pow_mod(int a, int n, int m)
{
    if(n == 0) return 1;
    int x = pow_mod(a, n/2, m);
    long long ans = (long long)x * x % m;
    if(n % 2 == 1) ans = ans *a % m;
    return (int)ans;
}
int main(){
    int n,k;
    cin>>n>>k;
    int com=0;
    for(int i=0;i<n;i++) {
        cin>>ans[i];
        if(i!=0) if(ans[i]==ans[i-1]) com++;
        }
    if(ans[n-1]==ans[0]) com++;
    int dif=n-com;
    dp[1][dif]=k-2;
    dp[1][dif-1]=1;
    dp[1][dif+1]=1;
    for(int i=2;i<=dif;i++){
        for(int j=0;j<=2*dif;j++){
            if(j-1>=0) dp[i][j]+=dp[i-1][j-1];
            dp[i][j]%=mo;
            if(j+1<=2*dif) dp[i][j]+=dp[i-1][j+1];
            dp[i][j]%=mo;
            dp[i][j]+=dp[i-1][j]*(k-2);
            dp[i][j]%=mo;
        }
        }
    long long ans=0;
    for(int i=dif+1;i<=2*dif;i++) {ans+=dp[dif][i];ans%=mo;}
    ans*=pow_mod(k,com,mo);
    ans%=mo;
    cout<<ans;
    }

 新人请多多包含。