D. Binary String To Subsequences (div3) (字符串思维)

D. Binary String To Subsequences

题意: 给一个01串,要求拆分成尽量少的子序列,每个子序列满足01交替,例如:101010或者010101,输出子序列个数 及原串中每个字符所属的子序列编号。
思路: 由于字符串长度2e5,所以考虑 O(n) 或者 O(nlogn) 的做法。
我们考虑如何使得子序列最少:在遍历时当前位为1时,那么如果之前有以0结尾的子序列,那直接接上去,并且此子序列的末尾状态也改变了,否则重新另开一个子序列。
那么就可以用两个队列,动态的维护是否有以 0 或 1 结尾的子序列,并且记录信息。
Code:

#include<bits/stdc++.h>
#define F first
#define S second
using namespace std;
typedef long long LL;

char s[200010];
int vis[200010];

int main()
{
	int T;cin>>T;
	while(T--){
		int n;
		scanf("%d%s",&n,s+1);
		queue<int> q0,q1;
		for(int i=0;i<=n;i++) vis[i]=0;
		int cnt=0;
		for(int i=1;i<=n;i++){
			if(s[i]=='0'){
				if(!q1.empty()){	//之前有以1结尾的子序列
					int t = q1.front();
					vis[i] = vis[t];
					q1.pop();
					q0.push(i);
				}else {
					q0.push(i);		//之前没有以1结尾的子序列,新开一个
					vis[i] = ++cnt;
				}
			}else {
				if(!q0.empty()){ 
					int t = q0.front();
					vis[i] = vis[t];
					q0.pop();
					q1.push(i);
				}else {
					q1.push(i);
					vis[i] = ++cnt;
				}
			}
		}
		printf("%d\n",cnt);
		for(int i=1;i<=n;i++) printf("%d ",vis[i]);
		printf("\n");
	}
	return 0;
}
©️2020 CSDN 皮肤主题: 我行我“速” 设计师:Amelia_0503 返回首页