#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
int n;
cin >> n;
vector<pair<int, int>> sumy;
for(int i=0;i<n;i++){
int x;
cin >> x;
sumy.push_back(make_pair(x, i));
}
sort(sumy.begin(), sumy.end(), [](pair<int, int> p1, pair<int, int> p2){
return p1.first < p2.first;
});
int index = 0;
long long sum = 0;
long long tempSum = 0;
for(int i=0;i<n;i++){
if(sum<=sumy[i].first){
index = i;
}
sum += tempSum;
tempSum = 0;
while(i+1<n&&sumy[i].first==sumy[i+1].first){
tempSum += sumy[i].first;
i++;
}
if(sum!=0){
sum+=tempSum;
tempSum = 0;
}
sum += sumy[i].first;
}
vector<bool> ans(n);
for(int i=0;i<n;i++){
if(i<index || sumy[0].first == sumy[n-1].first){
ans[sumy[i].second]=false;
}else {
ans[sumy[i].second]=true;
}
}
for(int i=0;i<n;i++){
cout << (ans[i] ? "T" : "N");
}
cout << endl;
}
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 44 45 46 47 48 49 50 51 | #include<iostream> #include<vector> #include<algorithm> using namespace std; int main(){ int n; cin >> n; vector<pair<int, int>> sumy; for(int i=0;i<n;i++){ int x; cin >> x; sumy.push_back(make_pair(x, i)); } sort(sumy.begin(), sumy.end(), [](pair<int, int> p1, pair<int, int> p2){ return p1.first < p2.first; }); int index = 0; long long sum = 0; long long tempSum = 0; for(int i=0;i<n;i++){ if(sum<=sumy[i].first){ index = i; } sum += tempSum; tempSum = 0; while(i+1<n&&sumy[i].first==sumy[i+1].first){ tempSum += sumy[i].first; i++; } if(sum!=0){ sum+=tempSum; tempSum = 0; } sum += sumy[i].first; } vector<bool> ans(n); for(int i=0;i<n;i++){ if(i<index || sumy[0].first == sumy[n-1].first){ ans[sumy[i].second]=false; }else { ans[sumy[i].second]=true; } } for(int i=0;i<n;i++){ cout << (ans[i] ? "T" : "N"); } cout << endl; } |
English