AtCoder Beginner Contest 214 F - Substrings(subsequence DP)

---

点击跳转

题意：

给定一个长度为$n$的字符串，每次不能选择相邻的字符，求子序列个数。
$1<=n<=2e5$

思路

• 比较容易想到的$dp$状态为:$dp[i]$表示从$[1,i]$中选并且$s[i]$必选的方案数
• 转移为$dp[i]={\sum }_{j=0}^{i-1}dp[j]$;
• 这样很容易会造成计数重复，比如$abacba$，当枚举到最后一个$a$的时候，可以跟第一个$a$组成$aa$，也可以跟第三个$a$组成$aa$。
• 会发现只需要维护到离$i$最近的最大的$j$就可以了，记为$k$。因为$k$之前的子序列想要用$s_i$的话，就可以用$s_k$替代，这样就不会重复了。
• 转移为$dp[i]={\sum }_{j=0}^{k}dp[j]$;
• 如果$k==0$的话，说明前面没有相同的字符，从起始开始转移就好了；
• 时间复杂度：看起来感觉是$O(n^2)$的，但最多只有$26$种字母，所以第二层循环最多枚举到$26$，时间复杂度$O(26n)$
• 再来考虑本题，加上了不能选相邻条件的限制，只需要算的时候将答案向后平移即可：
• $dp[i+2]={\sum }_{j=i}^{j>=k}dp[j]$
• 实现起来有很多细节
• 比如转移方程是将$dp[i]$的答案累加到了$dp[i+2]$上，计算答案的时候要从$2$到$n+1$，相当于整体向后平移
• 由于$j==i$的时候一定符合题意(选$i$和$i+2$)，所以判断跳出循环的条件是在计算更新答案之后的，就要注意条件的判断了；
• $j==0$的时候说明到达字符串的开始位置了；$s[j-1]==s[i]$说明下一位两个字符相同，也就是$j-1==k$，可以跳出了。

代码：

// Problem: F - Substrings
// Contest: AtCoder - AtCoder Beginner Contest 214
// URL: https://atcoder.jp/contests/abc214/tasks/abc214_f
// Memory Limit: 1024 MB
// Time Limit: 2000 ms
// Author:Cutele
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>PLL;
typedef pair<int, int>PII;
typedef pair<double, double>PDD;
#define I_int ll
inline ll read(){ll x = 0, f = 1;char ch = getchar();while(ch < '0' || ch > '9'){if(ch == '-')f = -1;ch = getchar();}while(ch >= '0' && ch <= '9'){x = x * 10 + ch - '0';ch = getchar();}return x * f;}

inline void write(ll x){if (x < 0) x = ~x + 1, putchar('-');if (x > 9) write(x / 10);putchar(x % 10 + '0');}

#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)

ll ksm(ll a, ll b,ll mod){ll res = 1;while(b){if(b&1)res=res*a%mod;a=a*a%mod;b>>=1;}return res;}

const int maxn=2e5+7,mod=1e9+7;

int dp[maxn],n;

int main(){
	string s;cin>>s;
	n=s.size();
	dp[0]=1;
	for(int i=0;i<n;i++)
		for(int j=i;;j--){
			dp[i+2]=(dp[i+2]+dp[j])%mod;
			if(s[j-1]==s[i]) break;
			if(j==0) break;
		}
	ll ans=0;
	rep(i,2,n+1) ans=(ans+dp[i])%mod;
	write(ans);
	return 0;
}