题目链接
题意:给定一个01字符串,将其分成n个子序列,使得每个子序列都没有连续的0或1,问n的最小值和每一个字符所属的序列编号。
思路:使用两个队列
q
0
,
q
1
q0,q1
q0,q1分别存储0和1的下标(同时两个队列的长度之差也体现了0和1的数量差),遍历字符串:
- 遇见’0’时,若q1为空则必须开辟一个新的子序列来存储这个多出来的’0’,若q1不为空,它就直接放到上一个1所在的子序列的后面,也就是子序列编号相同。
- 遇见’1’时同理
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define rep(i, a, b) for(int i=a;i<=b;i++)
#define frep(i, a, b) for(int i=a;i>=b;i--)
const int N = 1E5 + 10;
ll n;
int main() {
//freopen("1.txt","r",stdin);
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
int t;
cin >> t;
string str;
while (t--) {
cin>>n;
cin>>str;
queue<ll> q0,q1;
vector<ll> ans;
ll ind=1;
for(int i=0;i<n;i++){
if(str[i]=='0'){
if(q1.empty()) ans.push_back(ind++);
else ans.push_back(ans[q1.front()]),q1.pop();
q0.push(i);
}
else{
if(q0.empty()) ans.push_back(ind++);
else ans.push_back(ans[q0.front()]),q0.pop();
q1.push(i);
}
}
cout<<ind-1<<endl;
for(auto i:ans) cout<<i<<' ';cout<<endl;
}
return 0;
}
写在最后:这题一开始没有想到用类似队列或者栈的做法来解决,倒是想到了01数量差和记录上一个0、1出现的位置,但是后来还是没这样写;尝试的做法是从字符串的开始位置,子序列的编号先递增,再递减,…,以此类推,递增递减的切换条件是从0变1或者从1变0;但是这样写不仅复杂而且好像得不到正解,问题在于类似01100这种字符串它递增的时候难以判断是新开辟一个序列还是能够塞到原有的序列里头。总之这样写实在是太麻烦了,调的我头大,以后必须找更有效的解决方法而不是一直干一种**方法。