Codeforces Round #661 (Div. 3) D. Binary String To Subsequences 题解(思维)

题目链接

题目大意

给你一个长度为d的01串,要你把他分成若干个子序列,使得每个子序列都满足0和1交替出现,即没有相邻的0和相邻的1,要你求最少能分成多少个子序列,以及每个元素分别分给哪一个子序列

题目思路

比赛的时候想了好久都没想出来qwq,看了一下大佬的代码。其实这个思路谁都懂,如果现在元素是1看前面是否有0没有匹配,没有匹配则和原先那个0匹配,否则自己新成一个序列。这个元素为0也同理。这个实现就是记下没有被匹配的元素所处的子序列,然后运用队列的思想实现

代码

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define fi first
#define se second
#define debug printf(" I am here\n");
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const ll INF=0x3f3f3f3f3f3f3f3f;
const int maxn=2e5+5,inf=0x3f3f3f3f,mod=1e9+7;
const double eps=1e-10;
int d,ans[maxn];
char s[maxn];
signed main(){
    int _;scanf("%d",&_);
    while(_--){
        scanf("%d %s",&d,s+1);
        vector<int> vec[2];
        int cnt=0;
        for(int i=1;i<=d;i++){
            int x=s[i]-'0';
            if(vec[!x].empty()){
                vec[x].push_back(++cnt);
                ans[i]=cnt;
            }else{
                vec[x].push_back(vec[!x].back());
                ans[i]=vec[!x].back();
                vec[!x].pop_back();
            }
        }
        printf("%d\n",cnt);
        for(int i=1;i<=d;i++){
            printf("%d%c",ans[i],i==d?'\n':' ');
        }
    }
    return 0;
}

你可能感兴趣的:(思维)