oj.uz

답안 #617577

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
617577 2022-08-01T12:16:45 Z drdilyor Diversity (CEOI21_diversity) C++17
4 / 100
 14 ms 2644 KB 
#include <bits/stdc++.h>
#ifdef ONPC
#define cerr cout
    #include "t_debug.cpp"
#else
    #define debug(...) 42
#endif
#define sz(a) ((int)(a).size())
using namespace std;
using ll = long long;
const int INF = 1e9;
const ll INFL = 1e18;
const int RANDOM = chrono::high_resolution_clock::now().time_since_epoch().count();
mt19937 rng(RANDOM);
template<typename T, typename U> istream& operator>>(istream& is, pair<T, U>& p) { return is >> p.first >> p.second; }
const int N = 3e5, LOGN = 17, MOD = 1e9+7;

ll brute(vector<int> arr) {
    int n = sz(arr);
    ll res = INFL;
    do {
        ll cur = 0;
        for (int i = 0; i < n; i++) {
            set<int> values;
            for (int j = i; j < n; j++) {
                values.insert(arr[j]);
                cur += sz(values);
            }
        }
        if (cur < res) {
            res = cur;
            debug(cur, arr);
        }
    } while (next_permutation(arr.begin(), arr.end()));
    return res;
}
int solve() {
    int n, q;
    if (!(cin >> n >> q)) {
        return 1;
    }
    vector<int> arr(n);
    for (auto& i : arr) {cin >> i;}

    vector<int> count(N+1);
    for (int i : arr) count[i]++;

    vector<pair<int,int>> bycount;
    for (int i = 0; i <= N; i++) {
        if (count[i]) bycount.emplace_back(count[i], i);
    }
    sort(bycount.begin(), bycount.end());

    vector<int> brr;
    for (int i = 0; i < sz(bycount); i += 2) {
        brr.push_back(bycount[i].first);
    }
    for (int i = sz(bycount) - 1 - sz(bycount)%2; i >= 0; i-= 2){ 
        brr.push_back(bycount[i].first);
    }
    debug(brr);

    n = sz(brr);
    ll res = 0;

    for (int i = 0; i < n;i++) {
        for (int j = i+1; j < n; j++) {
            res += brr[i] * brr[j] * (j - i + 1);
        }
        res += brr[i] * (brr[i] + 1) / 2;
    }
    cout << res << '\n';
    
    return 0;
}

signed main() {
    cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    #ifdef ONPC
    t = 10000;
    #endif
    while (t-- && cin) {
        if (solve()) break;
        #ifdef ONPC
            cout << "____________________" << endl;
        #endif
    }
    return 0;
}

/*
     █████               █████  ███  ████                               
    ▒▒███               ▒▒███  ▒▒▒  ▒▒███                               
  ███████  ████████   ███████  ████  ▒███  █████ ████  ██████  ████████ 
 ███▒▒███ ▒▒███▒▒███ ███▒▒███ ▒▒███  ▒███ ▒▒███ ▒███  ███▒▒███▒▒███▒▒███
▒███ ▒███  ▒███ ▒▒▒ ▒███ ▒███  ▒███  ▒███  ▒███ ▒███ ▒███ ▒███ ▒███ ▒▒▒ 
▒███ ▒███  ▒███     ▒███ ▒███  ▒███  ▒███  ▒███ ▒███ ▒███ ▒███ ▒███     
▒▒████████ █████    ▒▒████████ █████ █████ ▒▒███████ ▒▒██████  █████    
 ▒▒▒▒▒▒▒▒ ▒▒▒▒▒      ▒▒▒▒▒▒▒▒ ▒▒▒▒▒ ▒▒▒▒▒   ▒▒▒▒▒███  ▒▒▒▒▒▒  ▒▒▒▒▒     
                                            ███ ▒███                    
                                           ▒▒██████                     
                                            ▒▒▒▒▒▒
*/

Compilation message

diversity.cpp: In function 'll brute(std::vector<int>)':
diversity.cpp:6:24: warning: statement has no effect [-Wunused-value]
    6 |     #define debug(...) 42
      |                        ^~
diversity.cpp:32:13: note: in expansion of macro 'debug'
   32 |             debug(cur, arr);
      |             ^~~~~
diversity.cpp: In function 'int solve()':
diversity.cpp:6:24: warning: statement has no effect [-Wunused-value]
    6 |     #define debug(...) 42
      |                        ^~
diversity.cpp:61:5: note: in expansion of macro 'debug'
   61 |     debug(brr);
      |     ^~~~~

Subtask #1
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 1 ms 1492 KB Output is correct
4 Correct 1 ms 1492 KB Output is correct
5 Correct 1 ms 1492 KB Output is correct
6 Correct 1 ms 1492 KB Output is correct
7 Correct 1 ms 1492 KB Output is correct
8 Correct 1 ms 1492 KB Output is correct
9 Correct 1 ms 1492 KB Output is correct
10 Correct 1 ms 1492 KB Output is correct

Subtask #2
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 2 ms 1492 KB Output is correct
4 Correct 8 ms 1880 KB Output is correct
5 Incorrect 14 ms 2644 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 2 ms 1492 KB Output is correct
4 Correct 8 ms 1880 KB Output is correct
5 Incorrect 14 ms 2644 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #4
0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 2 ms 1492 KB Output is correct
4 Correct 8 ms 1880 KB Output is correct
5 Incorrect 14 ms 2644 KB Output isn't correct
6 Halted 0 ms 0 KB -

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 1 ms 1492 KB Output is correct
4 Correct 1 ms 1492 KB Output is correct
5 Correct 1 ms 1492 KB Output is correct
6 Correct 1 ms 1492 KB Output is correct
7 Correct 1 ms 1492 KB Output is correct
8 Correct 1 ms 1492 KB Output is correct
9 Correct 1 ms 1492 KB Output is correct
10 Correct 1 ms 1492 KB Output is correct
11 Correct 1 ms 1492 KB Output is correct
12 Correct 1 ms 1492 KB Output is correct
13 Correct 2 ms 1492 KB Output is correct
14 Correct 8 ms 1880 KB Output is correct
15 Incorrect 14 ms 2644 KB Output isn't correct
16 Halted 0 ms 0 KB -

Subtask #6
0 / 36.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 1492 KB Output is correct
2 Correct 1 ms 1492 KB Output is correct
3 Correct 1 ms 1492 KB Output is correct
4 Correct 1 ms 1492 KB Output is correct
5 Correct 1 ms 1492 KB Output is correct
6 Correct 1 ms 1492 KB Output is correct
7 Correct 1 ms 1492 KB Output is correct
8 Correct 1 ms 1492 KB Output is correct
9 Correct 1 ms 1492 KB Output is correct
10 Correct 1 ms 1492 KB Output is correct
11 Correct 1 ms 1492 KB Output is correct
12 Correct 1 ms 1492 KB Output is correct
13 Correct 2 ms 1492 KB Output is correct
14 Correct 8 ms 1880 KB Output is correct
15 Incorrect 14 ms 2644 KB Output isn't correct
16 Halted 0 ms 0 KB -