0
点赞
收藏
分享

微信扫一扫

【DP】【子串】C. Unstable String

王栩的文字 2022-02-27 阅读 23

链接:
https://codeforces.com/problemset/problem/1535/C


状态表示:
f [ i ] f[i] f[i]:代表以 s [ i ] s[i] s[i]结尾的满足题意子串的数目
状态转移:
f [ i ] = f [ i − 1 ] + 1 f[i] = f[i - 1] +1 f[i]=f[i1]+1

pre决定了 [ p r e , i ] [pre,i] [pre,i]之间字符串的01状态,因为 s [ p r e ] s[pre] s[pre]已经确定

对转移进一步解释:

  • s [ i ] = ′ ? ′ s[i]='?' s[i]=?,该字符可以与前面的一个字符形成目标子串
  • s [ i ] s[i] s[i]为数字,需要找距离i最近的数字 s [ p r e ] s[pre] s[pre]进行判断,看 s [ p r e ] s[pre] s[pre] s [ i ] s[i] s[i]能否组成目标子串
    • 可以正常转移的条件:
      • i − p r e i-pre ipre为偶数且 s [ i ] = = s [ p r e ] s[i]==s[pre] s[i]==s[pre]
      • i − p r e i-pre ipre为奇数且 s [ i ] ! = s [ p r e ] s[i]!=s[pre] s[i]!=s[pre]
      • p r e = 0 pre=0 pre=0,代表还没有遇到数字

最后把以每一个字符结尾的结果加起来就可以了


#include<bits/stdc++.h>
using namespace std;
using ll = long long;

void solve()
{
	string s;
	cin >> s;
	int n = s.size();
	s = " " + s;
	
	vector<ll> f(n + 1, 0);
	
	int pre = 0;
	for(int i = 1; i <= n; i++)
	{
		if(s[i] == '?') f[i] = f[i - 1] + 1;
		else
		{
			int x = s[pre] - '0', y = s[i] - '0';
			if(pre == 0 || (i - pre) % 2 == 0 and x == y || (i - pre) & 1 and x != y) 
				f[i] = f[i - 1] + 1;
			else
				f[i] = i - pre;
			pre = i;
		} 		
	}
	
	ll res = 0;
	for(int i = 1; i <= n; i++)
		res += f[i];
	cout << res << "\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int t;
	cin >> t;
//	t = 1;
	while(t--) 
		solve();
	return 0;
}

举报

相关推荐

0 条评论