牛客编程巅峰赛S2第10场 第二题题解 优先队列 小根堆 十行代码
奇怪的排序问题
https://ac.nowcoder.com/acm/contest/10166/B
链接:https://ac.nowcoder.com/acm/contest/10166/B
来源:牛客网
题目描述
操场上有n个人排成一队,这n个人身高互不相同,可将他们的身高视为一个1到n的排列。
这时需要把队伍变成升序,也就是从矮到高排序。
每次可以选择一个人,让这个人和在他身后的人比高矮,如果比对方高,则交换位置并继续下一次比较,直到比对方矮或者已经在队尾。
现在给出数n和一个1到n的排列,求最少的选择次数,使队伍变为升序。
示例1
输入
4,[4,1,2,3]
返回值
1
备注:
n<=10^6
数据包含一个整数n和一个含有n个元素的数组,表示从队头到队尾的人的身高。
输出一个整数表示答案。
题解:
其实就是把高个调到后面,找到 存在后面“比自己矮的同学”的同学人数。
用小根堆从后往前遍历一遍,每次和堆顶比较,发现大的答案就加一,再压进堆,就行了。
会用priority_queue就很好写。
class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * @param n int整型 * @param a int整型vector * @return int整型 */ int wwork(int n, vector<int>& a) { // write code here int ans = 0; priority_queue<int, vector<int>, greater<int> > que; que.push(a[n-1]); for (int i = n-2; i >=0; i--) { if (a[i] > que.top()) { ans++; } que.push(a[i]); } return ans; } };