解题思路:我觉得https://blog.csdn.net/sakura_is_the_best/article/details/80156648讲得不错,但是有些细节我还是需要重新说一下。一开始我有一个疑问,像ggggggg这种用前面的不是很难受吗?因为上面博文中用的是abcdefgg来讨论的,只有末尾两个是相同的。其实是这样的,当s[i]==s[i-1]时,我们不管i-1之前是怎么取得,在这基础上来探究的。然后依次往前讨论。起决定作用的永远都是后面那几个字符串,因为他们会把前面和最后一个隔离开来。这也是为什么能用动态规划的原因。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
char s[1000009];
long long dp[1000009][4];
int main()
{
//freopen("t.txt","r",stdin);
scanf("%s",s+1);
int n=strlen(s+1);
for(int i=1;i<=n;i++)
dp[i][0]=1;
dp[1][1]=1;
for(int i=2;i<=n;i++)
{
if(s[i]==s[i-1])
dp[i][1]=dp[i-1][0]+dp[i-1][1]-1;
else
dp[i][1]=dp[i-1][0]+dp[i-1][1];
}
dp[2][2]=1;
for(int i=3;i<=n;i++)
{
if(s[i]==s[i-1])
dp[i][2]=dp[i-1][1]+dp[i-1][2]-dp[i-2][1];
else if(s[i]==s[i-2])
dp[i][2]=dp[i-1][1]+dp[i-1][2]-1;
else
dp[i][2]=dp[i-1][1]+dp[i-1][2];
}
dp[3][3]=1;
for(int i=4;i<=n;i++)
{
if(s[i]==s[i-1])
dp[i][3]=dp[i-1][3]+dp[i-1][2]-dp[i-2][2];
else if(s[i]==s[i-2])
dp[i][3]=dp[i-1][3]+dp[i-1][2]-dp[i-3][1];
else if(s[i]==s[i-3])
dp[i][3]=dp[i-1][3]+dp[i-1][2]-1;
else
dp[i][3]=dp[i-1][3]+dp[i-1][2];
}
printf("%lld\n",dp[n][3]+dp[n][2]+dp[n][1]+1);
return 0;
}