c_cpp 具有不同值的连续元素的数组的计数

Posted 2021-05-14

#include <bits/stdc++.h>
using namespace std;
#define mod 1000000007

long func(int n, int k, int x) {
    long dp[n];
    dp[0]= 0;
    dp[1]= 1;

    for (int i=2;i<n;i++) {
        dp[i]= ((k-1)*dp[i-2])%mod;
        dp[i]+= ((k-2)*dp[i-1])%mod;
        dp[i]%= mod;
    }
    return (x==1?((k-1)*dp[n-2])%mod:dp[n-1]);
}

int main() {
  ios_base
  int n,k,x;
  cin>>n>>k>>x;
  cout<< func(n,k,x)<<"\n";
}