Submission #779558

#	Time	Username	Problem	Language	Result	Execution time	Memory
779558		mindiyak	Arranging Shoes (IOI19_shoes)	C++14	10 / 100	111 ms	135068 KiB

shoes

#include "shoes.h"
#include <vector>
#include <deque>
#include <iostream>

#define pb push_back
#define mp make_pair
#define ll long long

using namespace std;

ll count_efffected(vector<ll> a,vector<ll> b){
	ll counter = 0,ans = 0;

	for(int i=0;i<a.size();i++){
		counter += a[i];
		a[i] = counter;
	}

	// for(int i=0;i<a.size();i++){
	// 	cout << a[i] << " ";
	// }cout << endl;

	// for(int i=0;i<b.size();i++){
	// 	cout << b[i] << " ";
	// }cout << endl;

	for(int i=0;i<b.size();i++){
		if(b[i] != -1){
			ans += a[b[i]] - a[i];
		}
	}

	return ans;
}

long long count_swaps(vector<int> s) {

	vector<deque<ll>> arr;
	vector<ll> effected_swap(s.size(),0);
	vector<ll> swap_pos(s.size(),-1);
	ll ans  = 0;

	deque<ll> a;
	for(int i=0;i<200003;i++){
		arr.pb(a);
	}
	
	for(int i=0;i<s.size();i++){
		// cout << s[i] << " (" << 100001-s[i] << ") " << arr[100001-s[i]].size();
		if(arr[100001-s[i]].size() != 0){
			// cout << " yes - " << arr[100001-s[i]][0] <<  "  " << (i-arr[100001-s[i]][0]) << " ";
			if(s[i] > 0){
				ans -= 1;
			}
			ans += i-arr[100001-s[i]][0];
			effected_swap[arr[100001-s[i]][0]+1] += 1;
			effected_swap[i] -= 1;
			swap_pos[i] = arr[100001-s[i]][0];
			arr[100001-s[i]].pop_front();
		}else{
			// cout << " no - " << s[i]+100001 << " ";
			arr[s[i]+100001].push_back(i);
		}
		// cout << ans << endl;
	}

	// cout << ans << endl;
	ans -= count_efffected(effected_swap,swap_pos);

	return ans;
}

Compilation message (stderr)

shoes.cpp: In function 'long long int count_efffected(std::vector<long long int>, std::vector<long long int>)':
shoes.cpp:15:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |  for(int i=0;i<a.size();i++){
      |              ~^~~~~~~~~
shoes.cpp:28:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |  for(int i=0;i<b.size();i++){
      |              ~^~~~~~~~~
shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:49:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |  for(int i=0;i<s.size();i++){
      |              ~^~~~~~~~~

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