#include<bits/stdc++.h> #define pb push_back #define int long long using namespace std; const int N=5e5+9; int n, a, ost, poprzednia, sumka, cur_sumka, maks, ile_maks, ile_zje[N], ile_razy; vector<int> v, org; unordered_map<int, char> mapa; unordered_map<int, int> kolejna, wystapienia; int32_t main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin>>n; if(n==1){ cout<<"T\n"; return 0; } for(int i=0; i<n; i++){ cin>>a; if(a>maks){ maks=a; ile_maks=1; } else if(a==maks) ile_maks++; v.pb(a); org.pb(a); } sort(v.begin(), v.end(), greater<int>()); ost=v.front(); poprzednia=v.front(); for(int u:v){ if(poprzednia==u){ kolejna[u]=ost; ile_razy++; } else{ kolejna[u]=poprzednia; wystapienia[poprzednia]=ile_razy; //cout<<poprzednia<<" wystepuje "<<ile_razy<<"\n"; ost=poprzednia; ile_razy=1; } poprzednia=u; } reverse(v.begin(), v.end()); /*for(int u:v){ cout<<"kolejna od "<<u<<" = "<<kolejna[u]<<"\n"; }*/ ost=0; for(int i=0; i<v.size(); i++){ int u=v[i]; if(u>ost){ sumka+=cur_sumka; cur_sumka=u; ost=u; } else cur_sumka+=u; ile_zje[i]=sumka; if(sumka>0 && wystapienia[u]) ile_zje[i]+=u*wystapienia[u]; //cout<<u<<" zje "<<ile_zje[i]<<"\n"; } for(int i=v.size()-1; i>=0; i--){ int u=v[i]; if((ile_zje[i]>kolejna[u] && mapa[kolejna[u]]=='T') || (ile_zje[i]>kolejna[u] && u==maks)) mapa[u]='T'; else mapa[u]='N'; } for(int u:org) cout<<mapa[u]; cout<<"\n"; return 0; }
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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | #include<bits/stdc++.h> #define pb push_back #define int long long using namespace std; const int N=5e5+9; int n, a, ost, poprzednia, sumka, cur_sumka, maks, ile_maks, ile_zje[N], ile_razy; vector<int> v, org; unordered_map<int, char> mapa; unordered_map<int, int> kolejna, wystapienia; int32_t main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin>>n; if(n==1){ cout<<"T\n"; return 0; } for(int i=0; i<n; i++){ cin>>a; if(a>maks){ maks=a; ile_maks=1; } else if(a==maks) ile_maks++; v.pb(a); org.pb(a); } sort(v.begin(), v.end(), greater<int>()); ost=v.front(); poprzednia=v.front(); for(int u:v){ if(poprzednia==u){ kolejna[u]=ost; ile_razy++; } else{ kolejna[u]=poprzednia; wystapienia[poprzednia]=ile_razy; //cout<<poprzednia<<" wystepuje "<<ile_razy<<"\n"; ost=poprzednia; ile_razy=1; } poprzednia=u; } reverse(v.begin(), v.end()); /*for(int u:v){ cout<<"kolejna od "<<u<<" = "<<kolejna[u]<<"\n"; }*/ ost=0; for(int i=0; i<v.size(); i++){ int u=v[i]; if(u>ost){ sumka+=cur_sumka; cur_sumka=u; ost=u; } else cur_sumka+=u; ile_zje[i]=sumka; if(sumka>0 && wystapienia[u]) ile_zje[i]+=u*wystapienia[u]; //cout<<u<<" zje "<<ile_zje[i]<<"\n"; } for(int i=v.size()-1; i>=0; i--){ int u=v[i]; if((ile_zje[i]>kolejna[u] && mapa[kolejna[u]]=='T') || (ile_zje[i]>kolejna[u] && u==maks)) mapa[u]='T'; else mapa[u]='N'; } for(int u:org) cout<<mapa[u]; cout<<"\n"; return 0; } |