Submission #74180

#	Time	Username	Problem	Language	Result	Execution time	Memory
74180		nvmdava	Ancient Books (IOI17_books)	C++17	0 / 100	5 ms	4472 KiB

books

#include "books.h"
#include <bits/stdc++.h>
using namespace std;
int k = 0;
int id[500001], l[500001], r[500001];
long long val[500001];
bool in[1000001];
int x;
vector<int> p;
void go(int v, long long c){
	if(in[v] == 1){
		val[k] = c;
		k++;
		return;
	}
	if(v <= x){
		l[k] = max(v, l[k]);
	}
	if(v >= x){
		r[k] = min(v, r[k]);
	}
	in[v] = 1;
	go(p[v], c + (long long)abs(v - p[v]));
}

long long minimum_walk(vector<int> p, int s) {
	::p = p;
	x = s;
	memset(r, 0x3f, sizeof r);
	memset(l, -1, sizeof l);
	for(int i = 0; i < p.size(); i++){
		if(in[i] || p[i] == i) continue;
		go(i, 0LL);
	}
	long long ans = 0;
	int loc = -1;
	for(int i = 0; i < k; i++){
		ans += val[i];
		loc = max(loc, r[i]);
		
	}
	ans += (long long) loc * 2;
	return ans;
}

Compilation message (stderr)

books.cpp: In function 'long long int minimum_walk(std::vector<int>, int)':
books.cpp:31:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < p.size(); i++){
                 ~~^~~~~~~~~~

• Subtask 1: 0 / 12
• Subtask 2: 0 / 10
• Subtask 3: 0 / 28
• Subtask 4: 0 / 20
• Subtask 5: 0 / 30