제출 #605770

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
605770		rrrr10000	Split the Attractions (IOI19_split)	C++14	18 / 100	904 ms	1048576 KiB

split

#include "split.h"

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<ll> vi;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef pair<ll,ll> P;
typedef vector<P> vp;
typedef vector<vp> vvp;
typedef vector<vvp> vvvp;
// typedef vector<bool> vb;
#define rep(i,n) for(ll i=0;i<(ll)(n);i++)
#define REP(i,k,n) for(ll i=(ll)(k);i<(ll)(n);i++)
#define all(a) a.begin(),a.end()
#define rsort(a) {sort(all(a));reverse(all(a));}
#define fi first
#define se second
#define pb emplace_back
template<class T> void out(T a){cout<<a<<endl;}
template<class T> void outv(T v){rep(i,v.size()){if(i)cout<<' ';cout<<v[i];}cout<<endl;}
template<class T> bool chmin(T&a,T b){if(a>b){a=b;return true;}return false;}
const ll inf=1001001001;
struct UF{
	vi par,sz;
	UF(ll n):par(n,-1),sz(n,1){}
	bool merge(ll a,ll b){
		a=root(a);b=root(b);
		if(a==b)return false;
		if(sz[a]<sz[b])swap(a,b);
		sz[a]+=sz[b];par[b]=a;
		return true;
	}
	ll root(ll i){
		if(par[i]==-1)return i;
		return root(par[i]);
	}
};

long double time_start;
long double gettime(){
	return (clock()-time_start)/CLOCKS_PER_SEC;
}
vector<int> find_split(int n, int A, int B, int C, vector<int> p, vector<int> q) {
	time_start=clock();
	ll k=min({A,B,C});
	vector<int> res(n);
	vvp g1(n);
	ll m=p.size();
	rep(i,m){
		g1[p[i]].pb(q[i],i);
		g1[q[i]].pb(p[i],i);
	}
	vi lk(n,inf),dep(n,-1),par(n);
	vector<bool> bridge(m,false);
	UF uf(n);
	function<void(ll,ll,ll)> dfs=[&](ll i,ll p,ll d){
		par[i]=p;dep[i]=d;
		for(auto x:g1[i])if(x.fi!=p){
			if(dep[x.fi]==-1){
				dfs(x.fi,i,d+1);
				chmin(lk[i],lk[x.fi]);
				if(lk[x.fi]>dep[i])bridge[x.se]=true;
				else uf.merge(i,x.fi);
			}
			else{
				chmin(lk[i],dep[x.fi]);
			}
		}
	};dfs(0,-1,0);
	vvp g2(n);
	// /*
	rep(i,m)if(bridge[i]){
		ll a=uf.root(p[i]),b=uf.root(q[i]);
		g2[a].pb(b,q[i]);g2[b].pb(a,p[i]);
	}
	// */
	/*
	rep(i,m){
		g2[p[i]].pb(q[i],-1);g2[q[i]].pb(p[i],-1);
	}
	*/
	queue<ll> que;
	vi deg(n);
	rep(i,n)deg[i]=g2[i].size();
	vvi gr1(n);
	rep(i,n)gr1[uf.root(i)].pb(i);
	/*
	rep(i,n)gr1[i].pb(i);
	*/
	rep(i,n)if(gr1[i].size()>0&&gr1[i].size()<k&&deg[i]==1)que.push(i);
	while(!que.empty()){
		ll t=que.front();que.pop();
		for(auto x:g2[t])if(gr1[x.fi].size()>0){
			if(gr1[x.fi].size()<gr1[t].size())swap(gr1[x.fi],gr1[t]);
			for(ll y:gr1[t])gr1[x.fi].pb(y);
			gr1[t].clear();
			deg[x.fi]--;
			if(deg[x.fi]==1&&gr1[x.fi].size()<k)que.push(x.fi);
		}
	}
	vi al;
	rep(i,n)if(gr1[i].size()>0)al.pb(i);
	auto sol=[&](ll sz,vector<bool> flag){
		ll i=0;
		while(!flag[i])i++;
		flag[i]=false;
		vi res;
		function<void(ll)> dfs_=[&](ll cur){
			if(res.size()==sz)return;
			res.pb(cur);
			for(auto x:g1[cur])if(flag[x.fi]){
				flag[x.fi]=false;
				dfs_(x.fi);
				if(res.size()==sz)return;
			}
		};dfs_(i);
		return res;
	};
	if(al.size()>1){
		ll t=0;
		while(deg[al[t]]!=1||gr1[al[t]].size()>n/2)t++;
		t=al[t];
		vi va,vb,vc;
		vector<bool> fa(n,false),fb(n,false),fc(n,true);
		for(ll x:gr1[t])fa[x]=true;
		rep(i,n)if(!fa[i])fb[i]=true;
		va=sol(k,fa);vb=sol(n-max({A,B,C})-k,fb);
		// assert(va.size()==k);
		// assert(vb.size()==n-max({A,B,C})-k);
		for(ll x:va)fc[x]=false;
		for(ll x:vb)fc[x]=false;
		rep(i,n)if(fc[i])vc.pb(i);
		// outv(va);outv(vb);outv(vc);
		if(A!=va.size())swap(va,vb);
		if(A!=va.size())swap(va,vc);
		if(B!=vb.size())swap(vb,vc);
		for(ll x:va)res[x]=1;
		for(ll x:vb)res[x]=2;
		for(ll x:vc)res[x]=3;
		return res;
	}
	ll rt=al[0];
	vector<bool> seen(n,false);
	rep(i,n)if(uf.root(i)==rt)seen[i]=true;
	vvi tmp(n);
	rep(i,n)if(uf.root(i)==rt){
		function<void(ll)> dfs=[&](ll cur){
			tmp[i].pb(cur);
			for(auto x:g1[cur])if(!seen[x.fi])dfs(x.fi);
		};dfs(i);
	}
	rep(i,n)if(tmp[i].size()>n-k)return res;
	while(gettime()<=0.9){
		vi ord(m);
		rep(i,m)ord[i]=i;
		random_shuffle(all(ord));
		UF uf(n);
		vvi g(n);
		for(ll i:ord)if(uf.merge(p[i],q[i])){
			g[p[i]].pb(q[i]);g[q[i]].pb(p[i]);
		}
		vi sub(n,1),par(n,-1),al;
		function<ll(ll,ll)> dfs=[&](ll i,ll p){
			al.pb(i);
			par[i]=p;
			for(ll x:g[i])if(x!=p){
				sub[i]+=dfs(x,i);
			}
			return sub[i];
		};dfs(0,-1);
		rep(i,n)if(sub[i]>=k&&sub[i]<=n-k){
			al=vi(0);dfs(i,par[i]);
			vi va,vb,vc;
			vector<bool> fa(n,false),fb(n,false),fc(n,true);
			for(ll x:al)fa[x]=true;
			rep(i,n)if(!fa[i])fb[i]=true;
			if(al.size()>n/2)swap(fa,fb);
			va=sol(k,fa);vb=sol(n-max({A,B,C})-k,fb);
			// assert(va.size()==k);
			// assert(vb.size()==n-max({A,B,C})-k);
			for(ll x:va)fc[x]=false;
			for(ll x:vb)fc[x]=false;
			rep(i,n)if(fc[i])vc.pb(i);
			// outv(va);outv(vb);outv(vc);
			if(A!=va.size())swap(va,vb);
			if(A!=va.size())swap(va,vc);
			if(B!=vb.size())swap(vb,vc);
			for(ll x:va)res[x]=1;
			for(ll x:vb)res[x]=2;
			for(ll x:vc)res[x]=3;
			return res;
		}
	}
}

컴파일 시 표준 에러 (stderr) 메시지

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:92:43: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   92 |  rep(i,n)if(gr1[i].size()>0&&gr1[i].size()<k&&deg[i]==1)que.push(i);
      |                              ~~~~~~~~~~~~~^~
split.cpp:100:37: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
  100 |    if(deg[x.fi]==1&&gr1[x.fi].size()<k)que.push(x.fi);
      |                     ~~~~~~~~~~~~~~~~^~
split.cpp: In lambda function:
split.cpp:111:17: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
  111 |    if(res.size()==sz)return;
      |       ~~~~~~~~~~^~~~
split.cpp:116:18: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
  116 |     if(res.size()==sz)return;
      |        ~~~~~~~~~~^~~~
split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:123:41: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  123 |   while(deg[al[t]]!=1||gr1[al[t]].size()>n/2)t++;
      |                        ~~~~~~~~~~~~~~~~~^~~~
split.cpp:136:7: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  136 |   if(A!=va.size())swap(va,vb);
      |      ~^~~~~~~~~~~
split.cpp:137:7: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  137 |   if(A!=va.size())swap(va,vc);
      |      ~^~~~~~~~~~~
split.cpp:138:7: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  138 |   if(B!=vb.size())swap(vb,vc);
      |      ~^~~~~~~~~~~
split.cpp:154:26: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
  154 |  rep(i,n)if(tmp[i].size()>n-k)return res;
      |             ~~~~~~~~~~~~~^~~~
split.cpp:179:16: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  179 |    if(al.size()>n/2)swap(fa,fb);
      |       ~~~~~~~~~^~~~
split.cpp:187:8: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  187 |    if(A!=va.size())swap(va,vb);
      |       ~^~~~~~~~~~~
split.cpp:188:8: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  188 |    if(A!=va.size())swap(va,vc);
      |       ~^~~~~~~~~~~
split.cpp:189:8: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  189 |    if(B!=vb.size())swap(vb,vc);
      |       ~^~~~~~~~~~~
split.cpp:49:10: warning: control reaches end of non-void function [-Wreturn-type]
   49 |  vvp g1(n);
      |          ^

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 0 / 22
• Subtask 4: 0 / 24
• Subtask 5: 0 / 36