結果

問題 No.1328 alligachi-problem
ユーザー chocorusk
提出日時 2020-12-25 00:50:17
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 685 ms / 2,000 ms
コード長 2,962 bytes
コンパイル時間 1,923 ms
コンパイル使用メモリ 141,256 KB
最終ジャッジ日時 2025-01-17 06:50:54
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#define popcount __builtin_popcount
using namespace std;
typedef long long ll;
typedef pair<int, int> P;

int main()
{
    int n; cin>>n;
    char c[200020], x[200020];
    int y[200020];
    vector<int> vr, vb;
    map<P, int> mp;
    for(int i=0; i<n; i++){
        cin>>c[i]>>x[i]>>y[i];
        if(c[i]==x[i]){
            mp[{x[i], y[i]}]++;
        }
        if(x[i]=='R') vr.push_back(i);
        else vb.push_back(i);
    }
    sort(vr.begin(), vr.end(), [&](int i, int j){
        if(y[i]!=y[j]) return y[i]<y[j];
        else return (x[i]==c[i])<(x[j]==c[j]);
    });
    sort(vb.begin(), vb.end(), [&](int i, int j){
        if(y[i]!=y[j]) return y[i]<y[j];
        else return (x[i]==c[i])<(x[j]==c[j]);
    });
    for(auto p:mp){
        if(p.second>1){
            cout<<"No"<<endl;
            return 0;
        }
    }
    reverse(vr.begin(), vr.end());
    reverse(vb.begin(), vb.end());
    vector<int> ans;
    int cr=0, cb=0;
    for(int i=0; i<n; i++){
        if(vr.empty()){
            ans.push_back(vb.back());
            vb.pop_back();continue;
        }
        if(vb.empty()){
            ans.push_back(vr.back());
            vr.pop_back();continue;
        }
        int p=vr.back(), q=vb.back();
        if(y[p]>cr && y[q]>cb){
            cout<<"No"<<endl;
            return 0;
        }
        if(y[p]>cr){
            vb.pop_back();
            ans.push_back(q);
            if(c[q]=='R') cr++;
            else cb++;
        }else if(y[q]>cb){
            vr.pop_back();
            ans.push_back(p);
            if(c[p]=='R') cr++;
            else cb++;
        }else{
            if(x[p]==c[p]){
                vr.pop_back();
                ans.push_back(p);
                if(c[p]=='R') cr++;
                else cb++;
            }else if(x[q]==c[q]){
                vb.pop_back();
                ans.push_back(q);
                if(c[p]=='R') cr++;
                else cb++;
            }else{
                cout<<"No"<<endl;
                return 0;
            }
        }
    }
    cr=0, cb=0;
    for(int i:ans){
        if(x[i]=='R'){
            if(cr!=y[i]){
                cout<<"No"<<endl;
                return 0;
            }
        }else{
            if(cb!=y[i]){
                cout<<"No"<<endl;
                return 0;
            }
        }
        if(c[i]=='R') cr++;
        else cb++;
    }
    cout<<"Yes"<<endl;
    for(int i=0; i<n; i++){
        cout<<ans[i]+1;
        if(i<n-1) cout<<" ";
    }
    cout<<endl;
    return 0;
}
0