結果
問題 | 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 |
ソースコード
#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; }