PAT 乙级 B1015 德才论 C++

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
struct student // 定义一个结构体 包含3个成员
{
    int id, de, cai;
};

bool cmp1(student a, student b) // 设计cmp1函数 将数组元素分类
{
    if (a.cai + a.de != b.cai + b.de)
    {
        return a.cai + a.de > b.cai + b.de;
    }
    else if (a.de != b.de)
    {
        return a.de >= b.de;
    }
    else
    {
        return a.id < b.id;
    }
}

int main()
{
    int n, min, max;
    cin >> n >> min >> max; // 输入考生人数 录取最低分数线 优先录取线
    int count = 0;          // 统计达到最低要求的考生
    struct student temp;
    vector<student> v[4];
    for (int i = 0; i < n; i++)
    {
        cin >> temp.id >> temp.de >> temp.cai; // 输入准考证号、德分、才分
        if (temp.de >= min && temp.cai >= min)
        {
            count++;                               // 达到最低分数线 则count++
            if (temp.de >= max && temp.cai >= max) // 根据题意 将考生分为四类  第一类 德才都达到优先录取线 按总分排序
            {                                      // 第二类 德达到优先录取线 才未达到 按德分排序
                v[0].push_back(temp);              // 第三类 德才都未达到优先录取线 但德大于等于才
            }                                      // 第四类 除去上面三类的考生
            else if (temp.de >= max && temp.cai <= max)
            {
                v[1].push_back(temp);
            }
            else if (temp.de <= max && temp.cai <= temp.de)
            {
                v[2].push_back(temp);
            }
            else
            {
                v[3].push_back(temp);
            }
        }
    }
    cout << count << endl;

    for (int i = 0; i < 4; i++)
    {
        sort(v[i].begin(), v[i].end(), cmp1); // 用sort函数和自定义的cmp1函数对数组排序
    }

    for (int i = 0; i < 4; i++) // v[0] v[1] v[2] v[3] 4个结构体数组
    {
        for (int j = 0; j < v[i].size(); j++)
        {
            cout << v[i][j].id << " ";
            cout << v[i][j].de << " ";
            cout << v[i][j].cai;
            cout << endl;
        }
    }

    system("pause");
    return 0;
}