[牛客]栗酱的文明2

对于每一个询问，输出一个数，即完美结局下，相互建立外交关系的国家数量。

2
5
2 2 2 2 2
10
8 6 5 9 2 7 10 3 3 9

2
6

对于第一个样例：
最多只能找到2个国家，使他们互相建立外交关系。
对于第二个样例：
第1,2,4,6,7,10个国家间可互相建立外交关系，最多数量为6。 

#include <bits/stdc++.h>
using namespace std;

// 定义比较函数，用于降序排序
bool cmp(int &a, int &b) {
    return a > b; // 如果a大于b，则返回true，表示a应该排在b之前
}

// 函数text处理每一组数据，计算并返回满足条件的土地数量
int text() {
    int gj; // 国家数（即每组数据的数量）
    cin >> gj;
    
    vector<int> tdarr(gj); // 初始化一个大小为gj的vector来存储土地数组

    // 输入土地数组的数据
    for (int j = 0; j < gj; ++j) {
        cin >> tdarr[j]; // 从标准输入读取每个国家的土地值，并存入tdarr
    }

    // 对土地数组进行降序排序，以便后续可以更高效地判断条件
    sort(tdarr.begin(), tdarr.end(), cmp);

    int count = 0; // 初始化计数器，用来记录满足条件的土地数量
    
    // 遍历排序后的土地数组，检查每个元素是否满足条件：元素值大于其索引位置
    for (int k = 0; k < gj; ++k) { 
        if (tdarr[k] > k) { // 如果当前元素值大于它的索引，则认为它满足条件
            count++; // 满足条件的土地数量加1
        } else {
            break; // 一旦遇到不满足条件的元素，后面的也不会满足，因此可以提前结束循环
        }
    }

    return count; // 返回该组中满足条件的土地数量
}

int main() {
    int n; // 组数，即有多少组数据需要处理
    cin >> n; // 从标准输入读取组数n

    // 外层循环遍历每一组数据
    for (int i = 0; i < n; ++i) {
        cout << text() << endl; // 调用text函数处理每组数据，并输出结果
    }
    
    return 0; // 程序正常结束
}

bool cmp(int &a, int &b) {
    return a > b;
}

int text() {
    int gj; // 国家数
    cin >> gj;
    vector<int> tdarr(gj); // 初始化vector

    // 输入土地数组
    for (int j = 0; j < gj; ++j) {
        cin >> tdarr[j];
    }

    // 排序
    sort(tdarr.begin(), tdarr.end(), cmp);

    int count = 0;
    // 计算满足条件的土地数量
    for (int k = 0; k < gj; ++k) { 
        if (tdarr[k] > k) {
            count++;
        } else {
            break; // 一旦遇到不满足条件的元素，后面的也不会满足，因此可以提前结束循环
        }
    }

    return count;
}

int main() {
    int n; // 组数
    cin >> n; // 从标准输入读取组数n

    for (int i = 0; i < n; ++i) {
        cout << text() << endl; // 调用text函数处理每组数据，并输出结果
    }
    
    return 0;
}