2022ccpc河北省赛9-优美字符串+DP

其实做的的时候没想到dp,看到题解恍然大悟 

/*
题意为寻找长度为n,含有m种字符串组成的字符串个数,要求该字符串不包含长度
大于不等于2的回文串。
思路:dp枚举dp[i][0/1/2],分别表示枚举到第i位以xyz,xxy,xyy形式结束的字符串数量 
*/
#include
using namespace std;
#define int long long 
const long long mod=1e9+7;
int dp[10000055][4];
signed main(){
	int n,m;
	cin>>n>>m;
	dp[3][0]=m*(m-1)%mod*(m-2)%mod;//最后三位以xyz形式
	//注意这里在中间要模一次是因为防止爆longlong,亲测有效 
	dp[3][1]=m*(m-1)%mod;//最后三位以xxy形式 
	dp[3][2]=m*(m-1)%mod;//最后三位呈现xyy形式 
	for(int i=4;i<=n;i++){
		dp[i][0]=(dp[i-1][0]*(m-2)%mod+dp[i-1][1]*(m-2))%mod;
		dp[i][1]=dp[i-1][2]*(m-2)%mod;
		dp[i][2]=(dp[i-1][0]+dp[i-1][1])%mod;
	}
	if(n==1){
		cout<

你可能感兴趣的:(c++,算法,开发语言)