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

最新推荐文章于 2024-06-04 05:41:15 发布

Knight840

最新推荐文章于 2024-06-04 05:41:15 发布

阅读量570

点赞数 1

文章标签： c++ 算法开发语言

本文链接：https://blog.csdn.net/qq_63108111/article/details/127519732

版权

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

/*
题意为寻找长度为n,含有m种字符串组成的字符串个数，要求该字符串不包含长度
大于不等于2的回文串。
思路:dp枚举dp[i][0/1/2],分别表示枚举到第i位以xyz，xxy，xyy形式结束的字符串数量 
*/
#include<bits/stdc++.h>
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<<m<<endl;
	} 
	else if(n==2){
		cout<<m*m%mod<<endl;
	}
	else {
		cout<<(dp[n][0]+dp[n][1]+dp[n][2])%mod<<endl;
	}
	return 0;
}