oj.uz

Submission #71301

# Submission time Handle Problem Language Result Execution time Memory
71301 2018-08-24T10:16:34 Z Kmcode Simurgh (IOI17_simurgh) C++14
51 / 100
 156 ms 10664 KB 
#include<bits/stdc++.h>
//#include "books.h"
using namespace std;

//#include "simurgh.h"
int count_common_roads(const std::vector<int>& r);

#define MAX 252*252
struct UF{
	int belong[MAX];
	int sz[MAX];
	int col[MAX];
	UF(){
		memset(belong,-1,sizeof(belong));
		for(int i=0;i<MAX;i++)sz[i]=1;
	}
	inline int root(int b){
		if(belong[b]==-1){
			return b;
		}
		return belong[b]=root(belong[b]);
	}
	void merge(int a,int b){
		a=root(a);
		b=root(b);
		if(a==b)return;
		if(a>b)swap(a,b);
		belong[a]=b;
		sz[b]+=sz[a];
	}
};
namespace solver{
	const int white=MAX-3;
	const int black=MAX-2;
	vector<pair<int,int> > v;
	vector<pair<int,int> > g[MAX];
	int id[MAX];
	int pr[MAX];
	int dep[MAX];
	UF uf;
	vector<int> road;
	vector<int> swa;
	
	inline void dfs(int b,int pr2=-1,int d=0){
		dep[b]=d;
			for(auto go:g[b]){
				if(go.first==pr2)continue;
				dfs(go.first,b,d+1);
				pr[go.first]=b;
				id[go.first]=go.second;
			}
	}
	int ini;
	int cn;
	int ask(){
		cn++;
		assert(cn<=30000);
		return count_common_roads(road);
	}
	void make_tree(){
		UF uf2;
		for(int i=0;i<v.size();i++){
			swa.push_back(-1);
			if(uf2.root(v[i].first)!=uf2.root(v[i].second)){
				uf2.merge(v[i].first,v[i].second);
				g[v[i].first].push_back(make_pair(v[i].second,i) );
				g[v[i].second].push_back(make_pair(v[i].first,i) );
				road.push_back(i);
				swa[i]=road.size()-1;
			}
		}
		ini=ask();
	}
	int query(int exp,int add){
		road[swa[exp]]=add;
		int z=ask();
		road[swa[exp]]=exp;
		return z-ini;
	}
	inline void check(int tree1,int tree2){
		int r1=uf.root(tree1);
		int r2=uf.root(tree2);
		if(r1==r2)return;
		if(r1==black||r1==white){
			if(r2==black||r2==white){
				return;
			}
		}
		int ret=query(tree1,tree2);
		if(ret==0){
			uf.merge(tree1,tree2);
			return;
		}
		if(ret>0){
			uf.merge(black,tree2);
			uf.merge(white,tree1);
			return;
		}
		uf.merge(black,tree1);
		uf.merge(white,tree2);
	}
	vector<int> mer[MAX];
	std::vector<int> find_roads(int n, std::vector<int> u1, std::vector<int> v1) {
		for(int i=0;i<u1.size();i++){
			v.push_back(make_pair(u1[i],v1[i]));
		}
		make_tree();
		if(ini==n-1){
			return road;
		}
		if(ini==0){
			for(int i=0;i<road.size();i++){
				uf.merge(road[i],white);
			}
		}
		dfs(0);
		for(int i=0;i<v.size();i++){
			if(dep[v[i].first]>dep[v[i].second])swap(v[i].first,v[i].second);
			if(pr[v[i].second]==v[i].first)continue;
			int a=v[i].first;
			int b=v[i].second;
			while(a!=b){
				if(dep[a]<dep[b]){
					check(id[b],i);
					b=pr[b];
				}
				else{
					check(id[a],i);
					a=pr[a];
				}
			}
		}
		vector<int> ans;
		UF uf2;
		for(int i=0;i<v.size();i++){
			if(uf.root(i)==black){
				ans.push_back(i);
				uf2.merge(v[i].first,v[i].second);
			}
		}
		for(int i=0;i<v.size();i++){
			mer[uf.root(i)].push_back(i);
		}
		for(int i=0;i<v.size();i++){
			if(i==white||i==black)continue;
			//mergeable
			if(mer[i].size()==0)continue;
			UF uf3;
			bool ok=false;
			for(int el:mer[i]){
				int a=v[el].first;
				int b=v[el].second;
				if(uf3.root(a)==uf3.root(b)){
					ok=true;
					break;
				}
				uf3.merge(a,b);
				if(uf2.root(a)==uf2.root(b)){
					ok=true;
					break;
				}
			}
			if(ok==false){
				for(int el:mer[i]){
					road.push_back(el);
					int a=v[el].first;
					int b=v[el].second;
					//uf2.merge(a,b);
				}
			}
		}
		assert(ans.size()>=n-1);
		return ans;
	}
	
}


std::vector<int> find_roads(int n, std::vector<int> u, std::vector<int> v) {
	return solver::find_roads(n,u,v);
}

Compilation message

simurgh.cpp: In function 'void solver::make_tree()':
simurgh.cpp:62:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
               ~^~~~~~~~~
simurgh.cpp: In function 'std::vector<int> solver::find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:104:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<u1.size();i++){
               ~^~~~~~~~~~
simurgh.cpp:112:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(int i=0;i<road.size();i++){
                ~^~~~~~~~~~~~
simurgh.cpp:117:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
               ~^~~~~~~~~
simurgh.cpp:135:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
               ~^~~~~~~~~
simurgh.cpp:141:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
               ~^~~~~~~~~
simurgh.cpp:144:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
               ~^~~~~~~~~
simurgh.cpp:166:10: warning: unused variable 'a' [-Wunused-variable]
      int a=v[el].first;
          ^
simurgh.cpp:167:10: warning: unused variable 'b' [-Wunused-variable]
      int b=v[el].second;
          ^
In file included from /usr/include/c++/7/cassert:44:0,
                 from /usr/include/x86_64-linux-gnu/c++/7/bits/stdc++.h:33,
                 from simurgh.cpp:1:
simurgh.cpp:172:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   assert(ans.size()>=n-1);
          ~~~~~~~~~~^~~~~

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 4856 KB correct
2 Correct 6 ms 4856 KB correct
3 Correct 7 ms 5044 KB correct
4 Correct 9 ms 5044 KB correct
5 Correct 6 ms 5048 KB correct
6 Correct 6 ms 5048 KB correct
7 Correct 6 ms 5048 KB correct
8 Correct 8 ms 5052 KB correct
9 Correct 6 ms 5052 KB correct
10 Correct 7 ms 5100 KB correct
11 Correct 6 ms 5100 KB correct
12 Correct 7 ms 5100 KB correct
13 Correct 7 ms 5100 KB correct

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 4856 KB correct
2 Correct 6 ms 4856 KB correct
3 Correct 7 ms 5044 KB correct
4 Correct 9 ms 5044 KB correct
5 Correct 6 ms 5048 KB correct
6 Correct 6 ms 5048 KB correct
7 Correct 6 ms 5048 KB correct
8 Correct 8 ms 5052 KB correct
9 Correct 6 ms 5052 KB correct
10 Correct 7 ms 5100 KB correct
11 Correct 6 ms 5100 KB correct
12 Correct 7 ms 5100 KB correct
13 Correct 7 ms 5100 KB correct
14 Correct 9 ms 5100 KB correct
15 Correct 9 ms 5124 KB correct
16 Correct 10 ms 5128 KB correct
17 Correct 8 ms 5128 KB correct
18 Correct 9 ms 5128 KB correct
19 Correct 9 ms 5144 KB correct
20 Correct 7 ms 5160 KB correct
21 Correct 8 ms 5164 KB correct
22 Correct 8 ms 5164 KB correct
23 Correct 7 ms 5164 KB correct
24 Correct 7 ms 5164 KB correct
25 Correct 8 ms 5164 KB correct
26 Correct 6 ms 5164 KB correct
27 Correct 9 ms 5164 KB correct
28 Correct 7 ms 5164 KB correct
29 Correct 6 ms 5164 KB correct
30 Correct 6 ms 5164 KB correct
31 Correct 6 ms 5164 KB correct
32 Correct 8 ms 5164 KB correct
33 Correct 9 ms 5164 KB correct

Subtask #3
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 4856 KB correct
2 Correct 6 ms 4856 KB correct
3 Correct 7 ms 5044 KB correct
4 Correct 9 ms 5044 KB correct
5 Correct 6 ms 5048 KB correct
6 Correct 6 ms 5048 KB correct
7 Correct 6 ms 5048 KB correct
8 Correct 8 ms 5052 KB correct
9 Correct 6 ms 5052 KB correct
10 Correct 7 ms 5100 KB correct
11 Correct 6 ms 5100 KB correct
12 Correct 7 ms 5100 KB correct
13 Correct 7 ms 5100 KB correct
14 Correct 9 ms 5100 KB correct
15 Correct 9 ms 5124 KB correct
16 Correct 10 ms 5128 KB correct
17 Correct 8 ms 5128 KB correct
18 Correct 9 ms 5128 KB correct
19 Correct 9 ms 5144 KB correct
20 Correct 7 ms 5160 KB correct
21 Correct 8 ms 5164 KB correct
22 Correct 8 ms 5164 KB correct
23 Correct 7 ms 5164 KB correct
24 Correct 7 ms 5164 KB correct
25 Correct 8 ms 5164 KB correct
26 Correct 6 ms 5164 KB correct
27 Correct 9 ms 5164 KB correct
28 Correct 7 ms 5164 KB correct
29 Correct 6 ms 5164 KB correct
30 Correct 6 ms 5164 KB correct
31 Correct 6 ms 5164 KB correct
32 Correct 8 ms 5164 KB correct
33 Correct 9 ms 5164 KB correct
34 Correct 136 ms 6736 KB correct
35 Correct 134 ms 6936 KB correct
36 Correct 105 ms 6936 KB correct
37 Correct 15 ms 6936 KB correct
38 Correct 149 ms 7320 KB correct
39 Correct 111 ms 7320 KB correct
40 Correct 105 ms 7320 KB correct
41 Correct 156 ms 7984 KB correct
42 Correct 136 ms 7984 KB correct
43 Correct 16 ms 7984 KB correct
44 Correct 12 ms 7984 KB correct
45 Correct 14 ms 7984 KB correct
46 Correct 10 ms 7984 KB correct
47 Correct 12 ms 7984 KB correct
48 Correct 8 ms 7984 KB correct
49 Correct 7 ms 7984 KB correct
50 Correct 8 ms 7984 KB correct
51 Correct 13 ms 7984 KB correct
52 Correct 11 ms 7984 KB correct
53 Correct 12 ms 7984 KB correct
54 Correct 12 ms 7984 KB correct
55 Correct 10 ms 7984 KB correct
56 Correct 12 ms 7984 KB correct
57 Correct 10 ms 7984 KB correct

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 7984 KB correct
2 Correct 7 ms 7984 KB correct
3 Incorrect 114 ms 10664 KB WA in grader: NO
4 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 4856 KB correct
2 Correct 6 ms 4856 KB correct
3 Correct 7 ms 5044 KB correct
4 Correct 9 ms 5044 KB correct
5 Correct 6 ms 5048 KB correct
6 Correct 6 ms 5048 KB correct
7 Correct 6 ms 5048 KB correct
8 Correct 8 ms 5052 KB correct
9 Correct 6 ms 5052 KB correct
10 Correct 7 ms 5100 KB correct
11 Correct 6 ms 5100 KB correct
12 Correct 7 ms 5100 KB correct
13 Correct 7 ms 5100 KB correct
14 Correct 9 ms 5100 KB correct
15 Correct 9 ms 5124 KB correct
16 Correct 10 ms 5128 KB correct
17 Correct 8 ms 5128 KB correct
18 Correct 9 ms 5128 KB correct
19 Correct 9 ms 5144 KB correct
20 Correct 7 ms 5160 KB correct
21 Correct 8 ms 5164 KB correct
22 Correct 8 ms 5164 KB correct
23 Correct 7 ms 5164 KB correct
24 Correct 7 ms 5164 KB correct
25 Correct 8 ms 5164 KB correct
26 Correct 6 ms 5164 KB correct
27 Correct 9 ms 5164 KB correct
28 Correct 7 ms 5164 KB correct
29 Correct 6 ms 5164 KB correct
30 Correct 6 ms 5164 KB correct
31 Correct 6 ms 5164 KB correct
32 Correct 8 ms 5164 KB correct
33 Correct 9 ms 5164 KB correct
34 Correct 136 ms 6736 KB correct
35 Correct 134 ms 6936 KB correct
36 Correct 105 ms 6936 KB correct
37 Correct 15 ms 6936 KB correct
38 Correct 149 ms 7320 KB correct
39 Correct 111 ms 7320 KB correct
40 Correct 105 ms 7320 KB correct
41 Correct 156 ms 7984 KB correct
42 Correct 136 ms 7984 KB correct
43 Correct 16 ms 7984 KB correct
44 Correct 12 ms 7984 KB correct
45 Correct 14 ms 7984 KB correct
46 Correct 10 ms 7984 KB correct
47 Correct 12 ms 7984 KB correct
48 Correct 8 ms 7984 KB correct
49 Correct 7 ms 7984 KB correct
50 Correct 8 ms 7984 KB correct
51 Correct 13 ms 7984 KB correct
52 Correct 11 ms 7984 KB correct
53 Correct 12 ms 7984 KB correct
54 Correct 12 ms 7984 KB correct
55 Correct 10 ms 7984 KB correct
56 Correct 12 ms 7984 KB correct
57 Correct 10 ms 7984 KB correct
58 Correct 6 ms 7984 KB correct
59 Correct 7 ms 7984 KB correct
60 Incorrect 114 ms 10664 KB WA in grader: NO
61 Halted 0 ms 0 KB -