This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=400005;
int ar[N];
ll get(int p){
ll ans=0;
while(p){
ans+=ar[p];
p-=p&(-p);
}
return ans;
}
void add(int p,ll val){
while(p<N){
ar[p]+=val;
p+=p&(-p);
}
}
long long count_swaps(std::vector<int> s) {
int n=s.size()/2;
vector<pair<int,int>>g[N];
for(int i=0;i<s.size();i++){
g[abs(s[i])].push_back({s[i],i});
add(i+1,1);
}
ll ans=0;
vector<pair<int,int>>ns;
for(int i=1;i<=n;i++){
sort(g[i].begin(),g[i].end());
int cnt=g[i].size()/2;
for(int j=0;j<cnt;j++){
int l=g[i][j].second;
int r=g[i][j+cnt].second;
if(l>r){
ans++;
swap(l,r);
}
ns.push_back({l+1,r+1});
}
}
//sort(ns.begin(),ns.end());
for(auto el:ns){
int l=el.first,r=el.second;
ans+=get(r-1)-get(l);
add(l,-1);add(r,-1);
}
return ans;
}
/*int main(){
vector<int>ar={2,1,-1,-2};
for(auto el:ar)cout<<el<<endl;
count_swaps(ar);
return 0;
}*/
Compilation message (stderr)
shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:24:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
24 | for(int i=0;i<s.size();i++){
| ~^~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |