Codeforces Round #521 (Div. 3) CDE

C

问删掉在数组中一个数后，剩下的数组中，是否存在一个数时其它所有数的和。

遍历每一个数，假设删掉它后，剩下的数中取一个最大值，看他的值是不是数组之和的一半，是的话就成立。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5+10;
ll a[N], n;
vector<ll> v;
int main() {
    cin >> n;
    ll sum = 0;
    priority_queue<ll> que;
    for(int i = 1; i <= n; i ++) cin >> a[i], sum += a[i], que.push(a[i]);
    int ans = 0;
    // cout << sum << ' ' << que.top() << endl;
    for(int i = 1; i <= n; i ++) {
        if(que.top() == a[i]) {
            que.pop();
            sum -= a[i];
            if(sum == que.top()*2) ans++, v.push_back(i);
            sum += a[i];
            que.push(a[i]);
        } else {
            sum -= a[i];
            if(sum == que.top()*2) ans++, v.push_back(i);
            sum += a[i];
        }
    }
    printf("%d\n",ans);
    for(auto x : v) printf("%lld ", x);printf("\n");
    return 0;
}

 

D 

有一个数组，可以选定一个子集，看最大能删除多少次，输出那个子集。

二分，检查当前最大删除次数时m时，是否成立。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5+10;
int a[N], n, k, x;
bool cmp(int x, int y) { return x > y;}
bool ok(int m) {
    int ans = 0;
    for(int i = 0; i < N; i ++) if(a[i])ans += a[i]/m;
    return ans >= k;
}
int main() {
    cin >> n >> k;
    for(int i = 1; i <= n; i ++) {
        cin >> x;
        a[x]++;
    }
    int l = 1, r = n/k, MAX = 0;
    while (l <= r) {
        int m = (l+r) >> 1;
        if(ok(m)) {
            MAX = max(MAX, m);
            l = m + 1;
        } else r = m-1;
    }
    int ans = 0;
    for(int i = 0; i < N; i ++) {
        for(int j = 0; j < a[i]/MAX; j ++) {
            if(ans == k) break;
            printf("%d%c",i," \n"[ans==k-1]);
            ans ++;
        }
    }
    return 0;
}

 

 

E

有很多主题，每个主题都有一些题目，选定一些主题，后面主题的问题数一定时前面的两倍，第一个主题的问题数随意，求最大的问题数。

 第一个主题的问从1个开始计算，一直到最多的问题。每次算出来问题总数后求最大值即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5+10;
int x, n, a[N], cnt;
map<int,int> mp;
bool vis[N];
int main() {
    cin >> n;
    for(int i = 1; i <= n; i ++) cin >> x, mp[x]++;
    for(auto m : mp) a[cnt++] = m.second;
    sort(a,a+cnt);
    int MAX = 0;
    for(int i = 1; i <= a[cnt-1]; i ++) {
        int x = i, ans = 0, l = 0, r = cnt;
        while(1) {
            int id = lower_bound(a+l,a+cnt,x)-a;
            if(id >= cnt) break;
            ans += x;
            l = id+1;
            x *= 2;
        }
        MAX = max(MAX, ans);
    }
    printf("%d\n",MAX);
    return 0;
}