買帽子

題目描述：

度度想去商場買一頂帽子，商場裏有N頂帽子，有些帽子的價格可能相同，度度想買一頂價格第三便宜的帽子，問第三便宜的帽子價格多少？

輸入描述：

首先輸入一個正整數N（N <= 50），接下來輸入N個數表示每頂帽子的價格（價格均是正整數，且小於等於1000）

輸出描述：

如果存在第三便宜的帽子，請輸出這個價格是多少，否則輸出-1

輸入例子：

10
10 10 10 10 20 20 30 30 40 40

輸出例子：

30

#include<iostream>
#include<vector>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;


int main()
{
    int n;
    cin>>n;
    if(n<3)
    {
        cout<<-1<<endl;
        return 0;
    }
    vector<int> v(n);
    for(int i=0;i<n;i++)
        cin>>v[i];
    sort(v.begin(),v.end());
    int j=0;
    bool flag=0;
    for(int i=1;i<n;i++)
    {
        if(v[j]!=v[i])
            v[++j]=v[i];
        if(j==2)
        {
            flag=1;
            break;
        }
    }
    if(flag)//若存在價格第三便宜的帽子
        cout<<v[2];
    else{
        cout<<-1<<endl;
        return 0;
    }
    return 0;

}

時間複雜度爲O（n）的解法：

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n, i, j;
    cin >> n;
    int a[50] = {0}, hash[1001] = {0};
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        hash[a[i]] = 1;
    }
    for (i = 1, j = 3; i < 1001; ++i) {
        if (hash[i] == 1)
            --j;
        if (j == 0) 
            break;
    }
    j > 0 ? cout << -1 : cout << i;
    return 0;
}