c++题解：小鱼之王

小鱼之王

题目描述

小贝一共养了 $n$ 条小鱼，其中第 $i$ 条小鱼的战斗力为 $a_i$ 。

小鱼之间可以相互战斗。如果小鱼 $x$ 和 小鱼 $y$ 的战斗力满足 $a_x>a_y$ ，那么小鱼 $x$ 可以战胜并吃掉 小鱼 $y$，并且吃掉后 小鱼 $x$ 的战斗力将会变为 $a_x+a_y$ 。

你可以任意安排小鱼之间战斗的顺序，直至仅剩下一条小鱼为止，剩下的小鱼将成为小鱼之王。请你编写程序计算一下，每条小鱼是否有成为小鱼之王的可能。

输入格式

输入的第一行，包含一个正整数 $n$。

输入的第二行，包含$n$ 个正整数，其中第 $i$ 个正整数 $a_i$ 表示第 $i$ 条小鱼的战斗力。

输出格式

输出仅一行，包含 $n$ 个字符表示每条小鱼的情况。其中用 $N$ 表示该条小鱼无法成为小鱼之王，用 $T$ 表示该条小鱼可以成为小鱼之王。

样例 #1

样例输入 #1

6
2 7 1 8 2 8

样例输出 #1

NTNTNT

样例 #2

样例输入 #2

3
5 4 4

样例输出 #2

TNN

提示

$1\le n\le 5\times 10^5$

$1\le a_i\le 10^9$

算法思想（二分）

根据题目描述，要找到哪些小鱼可以成为王者，可以通过找到满足成为王者的最小战力，如果其它小鱼高于这个战力，那么就可以成为王者，否则不行。根据题目描述，你可以任意安排小鱼之间战斗的顺序，因此可以将小鱼按照战斗力从小到大达排序，然后使用二分查找来解决。二分的条件就是，拥有该战力的小鱼能否成为王者。

注意：由于最后需要按照输入顺序输出结果，因此可以使用两个数组来处理。

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int n;
int a[N], b[N];
//检查拥有战力v的小鱼能否成为王者 
int check(int v)
{
	long long sum = v; //战斗力之和
    int flag = 0; //标记是否找到和自己一样战力的小鱼，这样处理可以跳过自己，防止重复累加
    for(int i = 1; i <= n; i ++)
    {
        if(flag == 0 && b[i] == v) { //跳过自己
            flag = 1;
            continue;
        }
        if(sum > b[i]) sum += b[i];
        else return 0;
    }
    return 1;
}
int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++){
        scanf("%d", a + i);
        b[i] = a[i];
    }
    sort(b + 1, b + n + 1);
    int L = 1, R = n;
    while(L < R)
    {
        int mid = (L + R) / 2;
        if(check(b[mid])) R = mid;
        else L = mid + 1;
    }
    int v = b[L]; //能够成为小鱼之王的最小战斗力
    for(int i = 1; i <= n; i ++)
        if(a[i] < v) putchar('N');
        else putchar('T');
    return 0;
}