Submission #810895

#TimeUsernameProblemLanguageResultExecution timeMemory
810895OAleksaArranging Shoes (IOI19_shoes)C++14
10 / 100
1082 ms10108 KiB
#include <bits/stdc++.h> #include "shoes.h" #define f first #define s second using namespace std; const int maxn = 1e5 + 69; set<int> l[maxn], r[maxn]; vector<int> f(maxn); void add(int v, int x) { for(int i = v;i < maxn;i += (i & -i)) f[i] += x; } int qry(int v) { int res = 0; for(int i = v;i >= 1;i -= (i & -i)) res += f[i]; return res; } long long count_swaps(vector<int> s) { int n = s.size(); vector<int> a(n + 1); for(int i = 0;i < n;i++) a[i + 1] = s[i]; for(int i = 1;i <= n;i++) { if(a[i] < 0) l[abs(a[i])].insert(i); else r[a[i]].insert(i); } long long ans = 0; for(int i = 1;i <= n;i += 2) { int x = a[i - qry(i)]; if(x < 0) { x = abs(x); auto u = *r[x].begin(); r[x].erase(u); l[x].erase(i); u += qry(u); add(i + 1, 1); add(u, -1); ans += (u - i - 1); } else { auto u = *l[x].begin(); l[x].erase(u); r[x].erase(i); u += qry(u); add(i, 1); add(u, -1); ans += (u - i); } } return ans; } // int main() // { // ios_base::sync_with_stdio(false); // cin.tie(0); // cout.tie(0); // int tt = 1; // //cin >> tt; // while(tt--) { // int n; // cin >> n; // vector<int> a(2 * n); // for(int i = 0;i < 2 * n;i++) // cin >> a[i]; // cout << count_swaps(a) << endl; // } // return 0; // }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...