Submission #341996

#	Time	Username	Problem	Language	Result	Execution time	Memory
341996		Marlov	고대 책들 (IOI17_books)	C++14	0 / 100	10 ms	17004 KiB

books

/*
Code by @marlov       
*/
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <vector>
#include <string>
#include <cmath>
#include <algorithm>
#include <iomanip>
#include <utility>
#include <set>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <queue>
#include <iterator>
#include <bitset>
using namespace std;
typedef long long ll;
typedef pair<long long,long long> ii;
 
#define maxN 1000005
#define INF 1000000000
 
long long N;
bool visited[maxN];
vector<int> child;
long long root;
long long cg=-1;
long long td[maxN];
long long mr[maxN];
vector<vector<ll> > grps;
 
void dfs(long long n){
	if(visited[n]) return;
	visited[n]=true;
	grps.back().push_back(n);
	td[cg]+=abs(n-child[n]);
	long long tr=n-root;
	if(abs(tr)<abs(mr[cg])) mr[cg]=tr;
	dfs(child[n]);
}
 
int minimum_walk(vector<int> p, int s){
	N=p.size();
	child=p;
	root=s;
	fill(visited,visited+maxN,false);
	fill(mr,mr+maxN,INF);
	cg=-1;
	grps.clear();
	fill(td,td+maxN,0);
	//if(p==vector<int> {3,2,1,0}) return 8;
	for(long long i=0;i<N;i++){
		if(!visited[i]){
			cg++;
			grps.push_back(vector<ll>());
			dfs(i);
		}
	}
	long long result=0;
	long long reach=0;
	long long ld=0;
	long long rd=0;
	for(long long i=0;i<=cg;i++){
		sort(grps[i].begin(),grps[i].end());
		bool conse=true;
		for(int j=1;j<grps[i].size();j++){
			if(grps[i][j]-1!=grps[i][j-1]) conse=false;
		}
		if(conse) result-=2*(*grps[i].begin());
	}
	for(long long i=0;i<=cg;i++){
		if(grps[i].size()==1) continue;
		result+=td[i];
		reach+=mr[i];
		//if(mr[i]<0) ld=min(ld,mr[i]);
		//if(mr[i]>=0) rd=max(rd,mr[i]);
	}
	return result+2*reach;
}
/*
int main() {
	ios_base::sync_with_stdio(0); cin.tie(0);
	//cout<<minimum_walk({0,2,3,1},0);
	//cout<<minimum_walk({0,1,2,3,4,6,5},0);
	vector<int> v = {0,1,2,3};
	do{
		cout<<minimum_walk(v,0)<<": ";
		for(int a:v) cout<<a<<" ";
		cout<<'\n';
	}while(next_permutation(v.begin(),v.end()));
	return 0;
}
*/

 
/* stuff you should look for
	* long long overflow, array bounds
	* special cases (n=1,n=0?)
	* do smth instead of nothing and stay organized
*/

Compilation message (stderr)

books.cpp: In function 'int minimum_walk(std::vector<int>, int)':
books.cpp:72:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |   for(int j=1;j<grps[i].size();j++){
      |               ~^~~~~~~~~~~~~~~
books.cpp:67:12: warning: unused variable 'ld' [-Wunused-variable]
   67 |  long long ld=0;
      |            ^~
books.cpp:68:12: warning: unused variable 'rd' [-Wunused-variable]
   68 |  long long rd=0;
      |            ^~

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