题目描述
排列好一队人,每个人身上有两个属性,一个是自己的身高height,一个是站在自己前面比自己高或者一样高的人的个数,打乱这队人原来的顺序,要求你根据给定的这一队人和他们的数据恢复他们原来的站位并输出,下面是原题描述以及例子,英语好的同学可以自行理解,
按题目的分类,这里要求使用到贪心算法,那么要怎么贪心才能不破坏最优解的情况下,一步一步求解呢?
根据观察,可以得到这么几个结论。
1. 如果一个人前面比自己高的人为0,那么有两种情况,要么他是最高的,要么他是第一个人。
2. 根据上一个结论,如果一个人身高是第二个人属性为0的中人最矮的,那么他必定是第一个人。
这样我们就找到了队伍的第一个人,也算是解决了一个子问题。之后的工作就很简单了,找到这个人之后,把他从原来的队伍中去掉,那么所有比他矮或者和他一样高的人,第二个属性都减1,如果是0则不变。然后问题又变成了找第二个属性为0的人中最矮的一个。以此类推,可以得到最终的结果。
我的代码与算法描述有点点不同,但是原理是一样的。
class Solution {
public:
static bool const compareTwoPare(const pair<int, int>& first, const pair<int, int>& second) {
if (first.second < second.second) {
return true;
} else if (first.second == second.second) {
if (first.first < second.first) {
return true;
} else {
return false;
}
} else {
return false;
}
}
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
sort(people.begin(), people.end(), compareTwoPare);
vector<pair<int, int> > result;
if (people.size() == 0) return result;
result.push_back(people[0]);
for (int i = 1; i < people.size(); ++i) {
int count = 0;
bool isadd = false;
for (int j = 0; j < result.size(); ++j) {
if (people[i].first <= result[j].first) {
if (count == people[i].second) {
isadd = true;
result.insert(result.begin() + j, 1, people[i]);
break;
}
++count;
}
}
if (!isadd) {
result.push_back(people[i]);
}
}
return result;
}
};