洛谷P9050题解

题目传送门

思路：

先来看看怎么判断一条鱼能不能活到最后。

我们可以遍历一遍数组，如果这条鱼就是当前考虑的鱼，那么就跳过（可以用结构体，存储每一条鱼的编号），如果这条鱼比当前的鱼轻，那么就sum（表示当前鱼的重量）加上这条鱼的重量，否则就return false；如果运行到了最后，就return true；

但是我们把每条鱼都考虑一遍，时间复杂度为，会超时。

那该怎么办呢？

如果一条鱼可以活到最后，那么比它重的鱼就一定能活到最后，比它轻的与不一定能活到最后，但是，可以确定的是，比它轻的鱼能不能活到最后不是乱序的，有一条比第一条相较于其它能活到最后的鱼都轻的，且能活到最后的鱼，比这条鱼轻的，很显然都不能活到最后，所以 ，鱼是否能活到最后满足单调性，所以我们可以用二分求解，时间复杂度：。

剩下的一些实现：

先排序。

我们已经通过二分找到了最轻的能活到最后的鱼，我们可以用一个标记数组，把能活到最后的鱼以及比它在数组中更靠后的鱼（因为都已经从小到大排完了序）标记为1，标记数组标记这条鱼的下标就是这条鱼在结构体中的编号，最后输出即可。

代码：

#include<iostream>
#include<algorithm>
#define int long long
using namespace std;
int n,b[500100];
struct node{
	int zhi,id;
}a[500100];
bool cmp(node c,node d){
	return c.zhi<d.zhi;
}bool check(int mid){
	int sum=a[mid].zhi;
	for(int i=1;i<=n;i++){
		if(a[i].id==a[mid].id){
			continue;
		}if(a[i].zhi>=sum){
			return false;
		}sum+=a[i].zhi;
	}return true;
}
signed main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i].zhi;
		a[i].id=i;
	}sort(a+1,a+1+n,cmp);
	int l=1,r=n,mid;
	while(l<=r){
		mid=(l+r)/2;
		if(check(mid)){
			r=mid-1;
		}else{
			l=mid+1;
		}
	}for(int i=l;i<=n;i++){
		b[a[i].id]=1;
	}for(int i=1;i<=n;i++){
		if(b[i]){
			cout<<"T";
		}else{
			cout<<"N";
		}
	}
	return 0;
} 

AC记录