Codeforces Round #479 (Div. 3) C. Less or Equal (排序,贪心)

Codeforces Round #479 (Div. 3)  C. Less or Equal  (排序,贪心)

  • 题意:有一个长度为\(n\)的序列,要求在\([1,10^9]\)中找一个\(x\),使得序列中恰好\(k\)个数满足\(\le x\).如果找不到\(x\),输出\(-1\).

  • 题解:先对这个序列排个序,然后首先要注意\(k=0\)的情况

    如果\(k=0\)并且序列中含有\(1\),那么我们无论如何都找不到比\(1\)小的数,输出\(-1\),如果不含\(1\),那么只要输出\(a[1]-1\)即可.

    如果\(k\ne 0\),那么我们要找前\(k\)个小的数(连续相等的数也算),所以我需要用桶来记录每个数出现的次数,然后遍历序列,累加每个数出现的次数,如果\(sum=k\),那么当前这个数就是我们要找的,如果\(sum>k\)的话,那么我们无论如何都找不到\(x\)(因为\(sum\)记录的是\(\le x\)的数目).

  • 代码:

    #include <iostream>
    #include <cstdio>
    #include <cstring>
    #include <cmath>
    #include <algorithm>
    #include <stack>
    #include <queue>
    #include <vector>
    #include <map>
    #include <set>
    #include <unordered_set>
    #include <unordered_map>
    #define ll long long
    #define fi first
    #define se second
    #define pb push_back
    #define me memset
    const int N = 1e6 + 10;
    const int mod = 1e9 + 7;
    using namespace std;
    typedef pair<int,int> PII;
    typedef pair<long,long> PLL; int n,k;
    int a[N];
    map<int,int> mp;
    int main() {
    ios::sync_with_stdio(false);cin.tie(0);
    cin>>n>>k;
    for(int i=1;i<=n;++i){
    cin>>a[i];
    mp[a[i]]++;
    } sort(a+1,a+1+n);
    if(k==0){
    if(mp[1]) puts("-1");
    else printf("%d\n",a[1]-1);
    return 0;
    }
    int cnt=0;
    for(auto w:mp){
    cnt+=w.se;
    if(cnt==k){
    printf("%d\n",w.fi);
    return 0;
    }
    if(cnt>k){
    puts("-1");
    return 0;
    }
    }
    puts("-1"); return 0;
    }
上一篇:***CI分页:为CodeIgniter写的分页类


下一篇:OCR OneNote