题意:你被给予了一个二进制字符串包含n个零和n个一。你的任务是分割这个字符串为最小的数量的子串,使得这些子串为'010101...'或者'101010...',输出每个字符属于的字符串编号。
分析:一开始想的是,字符个数为o(n),应该是一个线性的时间复杂度或者是o(nlogn)的时间复杂度,如果当前字符是'0',就查看是否之前存在过一个被分割的子串的结尾是'1',有的话就立马接上去,'0'的话也同理,如果接上去,那么就变成了一个新的末尾是相反字符的新子串,比如之前已经分割过一个'101',我们只能把'0'接在这个新子串的结尾,那么就变成'1010',产生了一个末尾是0的新子串的状态,原先的状态没了,暗示我们可以用队列处理,我们用pos记录第i个字符属于的字符串编号,扫描一遍每个字符,查看是否队列中存在和当前字符相反的末尾字符的子串,如果存在,那么就把当前字符的编号置为这个子串相同的编号。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 200005;
char str[N];
int pos[N];
int main()
{
int t;
scanf("%d", &t);
while (t--)
{
int n;
scanf("%d", &n);
scanf("%s", str + 1);
//01
queue<int> q1, q2;
int id = 0;
for (int i = 1; i <= n; ++i)
{
if (str[i] == '0')
{
if (q2.size() != 0)
{
int t = q2.front();
q2.pop();
pos[i] = pos[t];
q1.push(i);
}
else
{
pos[i] = ++id;
q1.push(i);
}
}
else
{
if (q1.size() != 0)
{
int t = q1.front();
q1.pop();
pos[i] = pos[t];
q2.push(i);
}
else
{
pos[i] = ++id;
q2.push(i);
}
}
}
printf("%d\n", id);
for (int i = 1; i <= n; ++i)
printf("%d ", pos[i]);
puts("");
}
return 0;
}