题目链接
题意:
现在定义一个子序列需要满足连续且不能含有相邻的0或1,给定一个由0和1组成的序列,求每个数字所在的子序列编号
题目分析:
维护两个队列,一个存放以1为结尾的子序列编号,另一个存方以0为结尾的子序列编号,从左到右模拟,如果为1,则寻找以0为结尾的子序列,标记编号,弹出0然后将1放入以1为结尾的队列中,如果不存在,新开一个。0的情况同理。
代码:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define pf push_front
#define int long long
#define debug(x) std::cerr << #x << " = " << (x) << std::endl
#define fast ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;
typedef long long ll;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
typedef set<int>::iterator SIT;
int dx[]={0,-1,0,1},dy[]={-1,0,1,0};
int gcd(int a,int b) {return b?gcd(b,a%b):a;}
int lcm(int a,int b) {return a/gcd(a,b)*b;}
const double eps=1e-6;
const int INF=0x3f3f3f3f,mod=998244353;
const int N=2e5+10;
char s[N];
int n;
int id[N];
signed main(){
fast;
int T;
cin>>T;
while(T--){
memset(id,0,sizeof(id));
queue<int>q0;//以0结尾
queue<int>q1;//以1结尾
cin>>n>>s+1;
int cnt=0;
for(int i=1;i<=n;i++){
if(s[i]=='0'){
if(q1.size()){//如果存在以1结尾
int t=q1.front();
id[i]=id[t];
q1.pop();
q0.push(i);
}
else{
q0.push(i);//新开一个以0结尾
id[i]=++cnt;
}
}
else{
if(q0.size()){
int t=q0.front();
id[i]=id[t];
q0.pop();
q1.push(i);
}
else{
q1.push(i);
id[i]=++cnt;
}
}
}
cout<<cnt<<endl;
for(int i=1;i<=n;i++)
cout<<id[i]<<' ';
cout<<endl;
}
return 0;
}