原题
度度熊想去商场买一顶帽子,商场里有N顶帽子,有些帽子的价格可能相同。
度度熊想买一顶价格第三便宜的帽子,问第三便宜的帽子价格是多少?
输入描述:
首先输入一个正整数N(N <= 50),接下来输入N个数表示每顶帽子的价格(价格均是正整数,且小于等于1000)
输出描述:
如果存在第三便宜的帽子,请输出这个价格是多少,否则输出-1
测试用例:
10
10 10 10 10 20 20 30 30 40 40
输出结果:
30
题解
解题思想:
采用最大堆的思想,用数组模拟一个元素个数为3的最大堆并赋初始值(初始值须大于题中要求的临界值),堆顶元素(即数组的第一个元素)永远保持为堆中最大,每插入一个元素,首先判断是否跟堆中元素相同,再判断是否比堆顶元素小,若无重复且小于堆顶元素,用其将堆顶元素覆盖,再调用堆调整算法。循环结束后,若数组第一个元素仍为初始值,则输出-1,反之输出数组第一个元素。该时间复杂度为O(N)。类似问题有:top k问题
代码:
#include <stdio.h>
void compare(int result[]){
if(result[1]>result[0]){
result[1]=result[1]^result[0];
result[0]=result[1]^result[0];
result[1]=result[1]^result[0];
}
if(result[2]>result[0]){
result[2]=result[2]^result[0];
result[0]=result[2]^result[0];
result[2]=result[2]^result[0];
}
}
int main(){
int N;
int price;
int result[3]={1001,1001,1001};
int i;
scanf("%d",&N);
for(i=0;i<N;i++){
scanf("%d",&price);
if(price!=result[0]&&price!=result[1]&&price!=result[2]&&price<result[0]){
result[0]=price;
compare(result);
}
}
if(result[0]==1001){
result[0]=-1;
}
printf("%d",result[0]);
}