QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#225611 | #4425. Cake | janix4000 | WA | 2366ms | 69620kb | C++14 | 3.1kb | 2023-10-24 21:00:26 | 2023-10-24 21:00:27 |
Judging History
answer
#include <bits/stdc++.h>
#include <vector>
template <typename... Ts>
void __print(Ts &&...ts) {}
#ifdef DEBUG
#include "print.hpp"
#endif // DEBUG
using namespace std;
#define all(x) x.begin(), x.end()
#define sz(x) ((int)(x).size())
using ll = long long;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using vs = vector<string>;
using vvs = vector<vs>;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
using vpi = vector<pi>;
///////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////
int _mergeSort(vi& arr, vi& temp, int left, int right);
int merge(vi& arr, vi& temp, int left, int mid, int right);
// This function sorts the
// input array and returns the
// number of inversions in the array
int mergeSort(vi& arr)
{
vi temp(arr.size());
return _mergeSort(arr, temp, 0, arr.size() - 1);
}
int _mergeSort(vi& arr, vi& temp, int left, int right)
{
int mid, inv_count = 0;
if (right > left) {
mid = (right + left) / 2;
inv_count += _mergeSort(arr, temp, left, mid);
inv_count += _mergeSort(arr, temp, mid + 1, right);
inv_count += merge(arr, temp, left, mid + 1, right);
}
return inv_count;
}
int merge(vi& arr, vi& temp, int left, int mid, int right)
{
int i, j, k;
int inv_count = 0;
i = left;
j = mid;
k = left;
while ((i <= mid - 1) && (j <= right)) {
if (arr[i] <= arr[j]) {
temp[k++] = arr[i++];
}
else {
temp[k++] = arr[j++];
inv_count = inv_count + (mid - i);
}
}
while (i <= mid - 1)
temp[k++] = arr[i++];
while (j <= right)
temp[k++] = arr[j++];
for (i = left; i <= right; i++)
arr[i] = temp[i];
return inv_count;
}
void solution() {
int n; cin >> n;
vvi in(2, vi(n));
vvi tg(2, vi(n));
for (size_t k = 0; k < 2; k++)
{
for (size_t i = 0; i < n; i++)
{
cin >> in[k][i];
}
}
for (size_t k = 0; k < 2; k++)
{
for (size_t i = 0; i < n; i++)
{
cin >> tg[k][i];
}
}
map<pi, vi> ids;
for (size_t i = 0; i < n; i++)
{
int a = tg[0][i];
int b = tg[1][i];
if(i % 2) swap(a, b);
ids[{a, b}].push_back(i);
}
vi res(n);
for (int i = n - 1; i >= 0; i--)
{
int a = in[0][i];
int b = in[1][i];
if(i % 2) swap(a, b);
auto it = ids.find({a, b});
if(it == ids.end() || it->second.empty()) {
res.clear();
break;
}
int idx = it->second.back();
it->second.pop_back();
res[i] = idx;
}
if(res.empty()) {
cout << -1 << endl;
} else {
cout << mergeSort(res) << endl;
}
}
int main(void) {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int qs = 1;
cin >> qs;
while (qs--) {
solution();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 2366ms
memory: 69620kb
input:
4 500000 471518156 319758862 812815356 520822448 129241996 461169933 796713727 608641317 281180101 953966756 749634941 274104949 996181952 88142916 998544672 125597509 991731126 974767231 338911715 674197249 167602044 682799026 226927279 703198907 216742488 8185420 94921423 690039818 859329736 45428...
output:
0 -1 445698416 -1855464059
result:
wrong answer 3rd lines differ - expected: '124999750000', found: '445698416'