用前缀和做的,50分,剩下TLE,求助
#include<bits/stdc++.h>
using namespace std;
long long s[500005];
int n;
struct node{
long long num,id;
}a[500005];
char c[500005];
bool cmp(node x,node y){
return x.num<y.num;
}
bool check(int i){
while(i<=n-1){
if(s[i]<=a[i+1].num){
return 0;
}
i++;
}
return 1;
}
int main(){
a[0].num=1000000001;
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i].num;
a[i].id=i;
}
sort(a+1,a+1+n,cmp);
for(int i=1;i<=n;i++){
s[i]=s[i-1]+a[i].num;
}
bool f=false;
for(int i=1;i<=n;i++){
if(a[i].num>a[i-1].num){
f=true;
}
if(a[i].num>=a[i-1].num&&check(i)&&f){
c[a[i].id]='T';
}
else{
c[a[i].id]='N';
}
}
for(int i=1;i<=n;i++){
cout<<c[i];
}
return 0;
}