#include "bits/stdc++.h" using namespace std; #define rep(i,a,b) for(int i=(a); i<(b); ++i) #define all(x) x.begin(),x.end() #define sz(x) int(x.size()) typedef long long ll; typedef unsigned long long ull; typedef vector<int> vi; typedef vector<vi> vvi; int main(){ cin.tie(NULL),cin.sync_with_stdio(false); int n; cin >> n; vector<bool> used(n), want = used, part = used; rep(i,0,n){ string s; cin >> s; want[i] = s == "TAK"; int t; cin >> t; part[i] = t < 2; } vi ans(20); int cc = 0; rep(i,0,n){ if (cc<10 and want[i]){ used[i] = 1; ans[cc++] = i; } } rep(i,0,n){ if (cc<20 and want[i] and part[i] and !used[i]) { used[i] = 1; ans[cc++] = i; } } for(auto& c : ans) cout << c+1 << ' '; cout << '\n'; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | #include "bits/stdc++.h" using namespace std; #define rep(i,a,b) for(int i=(a); i<(b); ++i) #define all(x) x.begin(),x.end() #define sz(x) int(x.size()) typedef long long ll; typedef unsigned long long ull; typedef vector<int> vi; typedef vector<vi> vvi; int main(){ cin.tie(NULL),cin.sync_with_stdio(false); int n; cin >> n; vector<bool> used(n), want = used, part = used; rep(i,0,n){ string s; cin >> s; want[i] = s == "TAK"; int t; cin >> t; part[i] = t < 2; } vi ans(20); int cc = 0; rep(i,0,n){ if (cc<10 and want[i]){ used[i] = 1; ans[cc++] = i; } } rep(i,0,n){ if (cc<20 and want[i] and part[i] and !used[i]) { used[i] = 1; ans[cc++] = i; } } for(auto& c : ans) cout << c+1 << ' '; cout << '\n'; } |