제출 #685036

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
685036		moonhero	Diversity (CEOI21_diversity)	C++14	0 / 100	1 ms	212 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;
        sort(res.begin(), res.end());
        reverse(res.begin() + res.size() / 2, res.end());
        if (res.size() > 20) abort();
        for (int i = 0; i < res.size(); i++) {
            ll k = 2, nwres = res[i];
            for (int j = i + 1; j < res.size(); j++) {
                nwres += k * res[j], k++;
            } h += nwres * res[i] - cont(res[i] - 1);
        }
        // while (next_permutation(res.begin(), res.end())) {
        //     ll x = 0;
        //     for (int i = 0; i < res.size(); i++) {
        //         ll k = 2, nwres = res[i];
        //         for (int j = i + 1; j < res.size(); j++) {
        //             nwres += k * res[j], k++;
        //         } x += nwres * res[i] - cont(res[i] - 1);
        //     } h = min(h, x);
        //     if (x == 93) {
        //         for (auto it : res) cout << it << ' ';
        //         cout << '\n';
        //     }
        // }
        cout << h << '\n';
    }
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

diversity.cpp: In function 'int main()':
diversity.cpp:27:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |         for (int i = 0; i < res.size(); i++) {
      |                         ~~^~~~~~~~~~~~
diversity.cpp:29:35: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |             for (int j = i + 1; j < res.size(); j++) {
      |                                 ~~^~~~~~~~~~~~

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