D. Binary String To Subsequences Codeforces Round #661 (Div. 3)

题目链接

思路：找每个0去匹配前面离他最近的1，每个1去匹配前面离他最近的0，如果1,0前面没有可以匹配的，那么就以自己独立开头开始匹配。

#include 
#include 
#include 
#include 
#include
#include
#include
#include
using namespace std;
typedef long long ll;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
const int mod=998244353;
const int N=2e6+10;
const int M=2e3+10;
const int inf=0x7f7f7f7f;
const int maxx=2e5+7;

ll gcd(ll a,ll b)
{
    return b==0?a:gcd(b,a%b);
}

ll lcm(ll a,ll b)
{
    return a*(b/gcd(a,b));
}

template <class T>
void read(T &x)
{
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
        if(c == '-')
            op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
        x = x * 10 + c - '0';
    if(op)
        x = -x;
}
template <class T>
void write(T x)
{
    if(x < 0)
        x = -x, putchar('-');
    if(x >= 10)
        write(x / 10);
    putchar('0' + x % 10);
}
ll qsm(int a,int b,int p)
{
    ll res=1%p;
    while(b)
    {
        if(b&1) res=res*a%p;
        a=1ll*a*a%p;
        b>>=1;
    }
    return res;
}

int r[N];
int a[N];

int main()
{

   int t;
   SIS;
   cin>>t;
   while(t--)
   {
       stack<int> q1,q0;
       int n;
       string s;
       cin>>n>>s;
       int len=0;
       for(int i=0;i<n;i++)
       {
           int x=s[i]-'0';
           if(x==0)
           {
               if(q1.empty())
               {
                   a[i]=++len;
                   q0.push(a[i]);
               }
               else
               {
                   a[i]=q1.top();
                   q1.pop();
                   q0.push(a[i]);
               }
           }
           else
           {
               if(q0.empty())
               {
                   a[i]=++len;
                   q1.push(a[i]);
               }
               else
               {
                   a[i]=q0.top();
                   q0.pop();
                   q1.push(a[i]);
               }
           }
          // len=max(len,a[i]);
       }
       //sort(a,a+n);
       cout<<len<<endl;
       for(int i=0;i<n;i++)
        cout<<a[i]<<' ';
       cout<<endl;

   }


    return 0;
}