SOJ 1063. Who's the Boss

题目大意：输入整数t，表示有t个测试样例。每个测试样例输入m和q，分别代表共有m个人和q个查询。接着有m行，每行代表一个人的信息，分别是id、年薪、身高。定义A的直接老板指的是m个人中，年薪比A大的且身高不低于A的身高人中年薪最低的人。定义A的下属是包括A的下属和A的下属的下属。接着q个查询每次输入一个id，查询该人的直接老板和下属的数量。

解题思路：排序递推。结构体Person包括id,salary,high属性。对年薪进行从小到大的排序，在目标id右侧找直接老板，在目标左边找下属数量。注意：从目标id往左，只要遇到身高高于目标id的身高就不能继续往左，因为此时更左侧的不是目标id的下属。

代码如下：

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;

struct Person {
    int id;
    int salary;
    int high;
    Person(int id_ = 0, int salary_ = 0, int high_ = 0) {
        id = id_;
        salary = salary_;
        high = high_;
    }
    bool operator<(const Person &other) const {
        if (salary < other.salary) {
            return true;
        } else {
            return false;
        }
    }
};


int main() {
    int t;
    cin >> t;
    while (t--) {
        int m, q;
        cin >> m >> q;
        vector<Person> persons;
        map<int, pair<int, int> > mp;
        int id, salary, high;
        for (int i = 0; i < m; i++) {
            cin >> id >> salary >> high;
            persons.push_back(Person(id, salary, high));
            mp[id] = pair<int, int>(salary, high);
        }

        sort(persons.begin(), persons.end());

        for (int i = 0; i < q; i++) {
            cin >> id;
            pair<int, int> p = mp[id];
            salary = p.first;
            high = p.second;

            // 二分查找  在当前根据salary排序完的队列中用二分查找，找到目标salary
            // 然后递推查找直系boss和下属的数量

            int begin = 0;
            int end = m;
            while (begin + 1 != end) {
                int mid = (begin + end) / 2;
                if (salary < persons[mid].salary) {
                    end = mid;
                } else {
                    begin = mid;
                }
            }

            int subordinate = 0;
            // begin 即 salary 对应的 person
            for (int i = begin-1; i >= 0; i--) {
                if (persons[i].high > high) break;
                subordinate++;
            }

            int immediateBoss = 0;
            for (int i = begin + 1; i < m; i++) {
                if (persons[i].salary > salary && persons[i].high >= high) {
                    immediateBoss = persons[i].id;
                    break;
                }
            }

            cout << immediateBoss << " " << subordinate << endl;

        }

    }

    return 0;
}