QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#105718 | #5507. Investors | MHJMBS | WA | 33ms | 3396kb | C++20 | 1.7kb | 2023-05-15 07:24:06 | 2023-05-15 07:24:09 |
Judging History
answer
#include "bits/stdc++.h"
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
#define fastio ios::sync_with_stdio(0), cin.tie(nullptr)
using namespace std;
using namespace __gnu_pbds;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using tiii = tuple<int,int,int>;
typedef tree<ll,null_type,less_equal<ll>, rb_tree_tag, tree_order_statistics_node_update> multiordered_set;
int main() {
fastio;
int t;
cin >> t;
while(t--) {
int n, k;
cin >> n >> k;
vector<ll> a(n);
for(ll &ai : a) cin >> ai;
while(k--) {
vector<int> dp(n+1, 0);
pll ans = {0,n};
multiordered_set msa, msd;
map<ll,int> occur_a;
for(int i = 0; i < n; i++) {
msa.insert(a[i]);
occur_a[a[i]]++;
}
for(int i = n-1; i >= 0; i--) {
int menores_depois = msd.order_of_key(a[i]);
msd.erase(msd.upper_bound(a[i]));
int maiores_antes = msa.size() - msa.order_of_key(a[i]) - occur_a[a[i]];
msa.erase(msa.upper_bound(a[i]));
occur_a[a[i]]--;
dp[i] = dp[i+1] + maiores_antes - menores_depois;
ans = max(ans, {dp[i],i});
}
for(int i = ans.second; i < n; i++) a[i] += ll(1e10);
}
multiordered_set ms;
map<ll,int> occur;
ll ans = 0;
for(int i = 0; i < n; i++) {
ans += i - ms.order_of_key(a[i]) - occur[a[i]];
ms.insert(a[i]);
occur[a[i]]++;
}
cout << ans << '\n';
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 2ms
memory: 3352kb
input:
2 6 1 4 5 6 2 2 1 6 2 4 5 6 2 2 1
output:
2 0
result:
ok 2 lines
Test #2:
score: -100
Wrong Answer
time: 33ms
memory: 3396kb
input:
349 6 2 2 1 2 1 2 1 48 12 42 47 39 39 27 25 22 44 45 13 5 48 38 4 37 6 24 10 42 38 12 37 31 19 44 6 29 17 7 12 7 26 35 24 15 9 37 3 27 21 33 29 34 20 14 30 31 21 48 12 1 43 17 46 17 39 40 22 25 2 22 12 4 11 38 12 4 11 1 5 39 44 37 10 19 20 42 45 2 45 37 20 48 34 16 41 23 18 13 44 47 21 29 4 23 18 16...
output:
1 174 108 406 0 10 2 0 143 150 204 0 0 0 1 0 0 0 0 0 0 0 0 375 3 0 0 2 0 0 0 0 0 0 1 0 3 0 0 1 0 0 2 0 0 1 15 0 0 0 0 0 0 0 0 2 0 2 0 0 71 522 0 0 239 4 0 1 0 0 3 0 0 0 0 0 0 0 4 0 0 0 0 0 0 0 0 0 0 0 0 0 9 0 0 0 3 0 0 0 0 0 0 0 39 1 53 0 0 0 0 1 156 5 0 0 0 44 0 0 0 0 0 2 0 0 0 0 0 0 1 0 0 0 0 0 0 ...
result:
wrong answer 2nd lines differ - expected: '18', found: '174'