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;
}