Submission #211148

#	Time	Username	Problem	Language	Result	Execution time	Memory
211148		vincent97198	Arranging Shoes (IOI19_shoes)	C++14	0 / 100	5 ms	384 KiB

shoes

#include <bits/stdc++.h>
#include "shoes.h"
#define N 100005

using namespace std;

int BIT[2*N];

inline int lowbit(int x)
{
	return x&(-x);
}

void add(int pos,int val)
{
	for(;pos<2*N;pos+=lowbit(pos))
		BIT[pos]+=val;
}

int sum(int pos)
{
	int ret=0;
	for(;pos>0;pos-=lowbit(pos))
		ret+=BIT[pos];
	return ret;
}

long long count_swaps(vector<int> S)
{
	map<long long,queue<int>> mp;
	
	long long ans=0;
	for(int i=0;i<S.size();++i){
		mp[S[i]].push(i);
		add(i+1,1);
	}
	for(int i=0;i<S.size();++i){
		if(mp[S[i]].empty() || mp[S[i]].front()!=i)	continue;
		mp[S[i]].pop();
		int t=mp[-S[i]].front();	mp[-S[i]].pop();
		if(S[i]>0)
			++ans;
		ans+=(sum(t-1)-sum(i+1));
		add(t+1,-1);
	}
	return ans;
}

Compilation message (stderr)

shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:33:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<S.size();++i){
              ~^~~~~~~~~
shoes.cpp:37:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<S.size();++i){
              ~^~~~~~~~~

• Subtask 1: 0 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20
• Subtask 6: 0 / 15