CSES - Datatähti Open 2021 - Results
Submission details
Task:Sorting
Sender:sandoval
Submission time:2021-01-30 06:38:48 +0200
Language:C++17
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.01 s1, 2details
#20.01 s2details
#30.01 s1, 2details
#40.01 s1, 2details

Code

#include <bits/stdc++.h>
using namespace std;
using int64 = long long;

int main() {
  ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
  map<int,int> cnt;
  int n,k; cin >> n >> k;
  for (int i = 0; i < n; ++i) {
    int x; cin >> x;
    ++cnt[x];
  }

  int l = 0;
  vector<int> rem;
  for (auto& x : cnt) {
    l += (x.second/2);
    x.second = x.second&1;
    if (x.second&1) {
      rem.push_back(x.first);
    }
  }
  if (l >= k) {
    cout << 0 << '\n';
    return 0;
  }
  k -= l;
  n = (int)rem.size();
  constexpr int64 INF = 1LL << 60;
  vector<vector<int64>> dp(1+n, vector<int64>(1+k, 0));
  for (int j = 1; j <= k; ++j) {
    for (int i = n; i >= 0; --i) {
      if (i >= n-1) {
        dp[i][j] = INF;
      } else {
        dp[i][j] = min(dp[1+i][j], dp[2+i][j-1]+rem[1+i]-rem[i]);
      }
    }
  }
  cout << dp[0][k] << '\n';
  return 0;
}

Test details

Test 1

Group: 1, 2

Verdict:

input
153
1
1
2
1 2
...

correct output
YES
YES
NO
NO
NO
...

user output
0

Test 2

Group: 2

Verdict:

input
1000
59
35 29 32 50 11 15 9 21 19 45 2...

correct output
YES
NO
YES
NO
YES
...

user output
0

Test 3

Group: 1, 2

Verdict:

input
720
6
1 6 4 5 2 3
6
6 3 2 1 5 4
...

correct output
YES
NO
NO
NO
YES
...

user output
0

Test 4

Group: 1, 2

Verdict:

input
1000
8
7 4 2 8 6 3 5 1
8
3 8 2 7 5 4 6 1
...

correct output
NO
NO
YES
NO
YES
...

user output
0