Submission #685047

#	Time	Username	Problem	Language	Result	Execution time	Memory
685047		moonhero	Diversity (CEOI21_diversity)	C++14	64 / 100	7026 ms	38924 KiB

diversity

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const long long N = 3e5 + 5;
long long a[N];
vector <long long> res;
long long cont (long long r) {
    return (r + 1) * (r) / 2;
}
signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    ll n, q; cin >> n >> q;
    map <ll, ll> ans, s;
    for (ll i = 1; i <= n; i++) {
        cin >> a[i];
        s[a[i]] = -1;
        ans[a[i]]++;
    } 
    while (q--) {
        ll l, r; cin >> l >> r;
        for (auto it : ans) res.push_back(it.second);
        ll h = 0, g = 0;
        sort(res.begin(), res.end());
        vector <int> rs(res.size());
        bool ok = 0;
        int lt = 0, rt = rs.size() - 1;
        for (auto it : res) {
            if (ok) rs[rt] = it, rt--, ok = 0;
            else rs[lt] = it, lt++, ok = 1;
        } 
        for (int i = 0; i < rs.size(); i++) res[i] = rs[i];
        ll pref = 0, sum = 0;
        for (int i = 0; i < res.size(); i++) {
            pref += res[i], sum += pref;
            // cout << sum << ' ' << res[i] << ' ' << i << '\n';
            h += res[i] * sum - cont(res[i] - 1);
        }
        cout << h << '\n';
    }
    return 0;
}

Compilation message (stderr)

diversity.cpp: In function 'int main()':
diversity.cpp:32:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |         for (int i = 0; i < rs.size(); i++) res[i] = rs[i];
      |                         ~~^~~~~~~~~~~
diversity.cpp:34:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   34 |         for (int i = 0; i < res.size(); i++) {
      |                         ~~^~~~~~~~~~~~
diversity.cpp:23:19: warning: unused variable 'g' [-Wunused-variable]
   23 |         ll h = 0, g = 0;
      |                   ^

• Subtask 1: 4 / 4
• Subtask 2: 10 / 10
• Subtask 3: 8 / 8
• Subtask 4: 16 / 16
• Subtask 5: 26 / 26
• Subtask 6: 0 / 36