Submission #302530

#	Time	Username	Problem	Language	Result	Execution time	Memory
302530		REALITYNB	슈퍼트리 잇기 (IOI20_supertrees)	C++14	100 / 100	263 ms	26384 KiB

supertrees

#include <bits/stdc++.h> 
#include "supertrees.h" 
#define all(a) a.begin(),a.end() 
using namespace std; 
const int mxn= 1e3+1  ; 
vector<int> adj[mxn] , adjj[mxn] , one  , mn ; 
void dfs(int i , vector<int>& a ){
	a[i]=1 ;
	for(int& x : adj[i]){
		if(a[x]) continue ; 
		dfs(x,a) ; 
	}
}
int cnt = 0 ; 
void dfss(int i , vector<int>& a){
	a[i] = 1 ;
	++cnt ;  
	if(adjj[i].size()==1) one.push_back(i) ; 
	for(int& x : adjj[i]){
		if(a[x]) continue ; 
		dfss(x,a) ; 
	}
}
vector<vector<int>> ppp; 
bool check(int i , vector<int>& a){
	a[i] =1 ; 
	for(int j=0;j<mn.size();j++){
		if(ppp[i][j]==2&&mn[j]!=2) return 1 ; 
		if(mn[j]==2&&ppp[i][j]!=2) return 1 ; 
	}
	bool b = 0 ; 
	for(int& x : adj[i]){
		if(a[x]) continue ; 
		b|=check(x,a) ; 
	}	
	return b ; 

}
int construct(vector<vector<int>> p){
	ppp= p ; 
	int n =p.size() ;
	vector<int> the(n) ; 
	for(int i=0;i<n;i++){
		bool flg = 0  ; 
		vector<int> x=p[i] ; 
		for(int j=0;j<i;j++){
			if(x[j]==1){
				adj[i].push_back(j) ; 
				flg = 1 ; 
				break ; 
			}
		}
		if(flg) continue ;
		//cout <<i << endl ; 
		the[i] =1 ; 
		for(int j=i+1;j<n;j++) if(x[j]==1) adj[i].push_back(j) ; 
	} 
	for(int i=0;i<n;i++){
		vector<int> vis(n) ; 
		dfs(i,vis) ;
		for(int j=0;j<n;j++){
			if(vis[j]==1&&p[i][j]==1) continue ; 
			if(vis[j]==0&&p[i][j]==0) continue ; 
			if(p[i][j]==2&&vis[j]==0) continue ; 
			return 0 ; 
		}
	}
	for(int i=0;i<n;i++){
		if(the[i]==0) continue ; 
		for(int j=i-1;j>-1;j--){
			if(the[j]==0) continue  ; 
			if(p[i][j]==2){
				adjj[i].push_back(j) ; 
				break ; 
			}
		}
		for(int j=i+1;j<n;j++){
			if(the[j]==0) continue ; 
			if(p[i][j]==2){
				adjj[i].push_back(j) ; 
				break ; 
			}
		}
	}

	map<vector<int>,int> pp ; 
	for(int i=0;i<n;i++){
		if(the[i]==0) continue ; 
	//	if(vis[i]) continue ; 
			vector<int> vis(n); 
		one.erase(one.begin(),one.end()) ;
		cnt= 0 ; 
		dfss(i,vis) ; 
		if(cnt==2) return 0 ; 
		sort(all(one)) ; 
		for(int j=0;j<n;j++){
			if(i==j) continue ; 
			/*if(vis[j]&&p[i][j]==0)return 0 ; 
			if(vis[j]==0&&p[i][j]) return 0 ; */
			if(the[j]==0) continue ; 
			if(vis[j]==1&&p[i][j]==2) continue ; 
			if(vis[j]==0&&p[i][j]==0) continue ; 
			if(vis[j]==0&&p[i][j]==1) continue ;
			return 0 ; 
			//cout << i << " " << j << endl ; 
		}
		if(one.size()==0) continue ; 
		if(one.size()==1) continue ; 
		if(one.size()>2) return 0 ; 
		if(pp[{one[0],one[1]}]) continue ;
			pp[{one[0],one[1]}]=1 ; 
			adjj[one[1]].push_back(one[0]) ; 
			adjj[one[0]].push_back(one[1])  ;
		}
	vector<int> viss(n) ; 
	for(int i=0;i<n;i++){
		if(viss[i]==1) continue ; 
		mn = p[i] ; 
		bool f = check(i,viss) ; 
		if(f) return 0 ; 
	}
	vector<vector<int>> ans ; 
	for(int i=0;i<n;i++) for(int& x : adjj[i]) adj[i].push_back(x) ; 
	for(int i=0;i<n;i++){
		vector<int> row(n) ; 
		for(int& x : adj[i]) row[x]= 1 ; 
		ans.push_back(row) ; 
	}
	build(ans) ; 
	return 1 ; 
}

Compilation message (stderr)

supertrees.cpp: In function 'bool check(int, std::vector<int>&)':
supertrees.cpp:27:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |  for(int j=0;j<mn.size();j++){
      |              ~^~~~~~~~~~

• Subtask 1: 11 / 11
• Subtask 2: 10 / 10
• Subtask 3: 19 / 19
• Subtask 4: 35 / 35
• Subtask 5: 21 / 21
• Subtask 6: 4 / 4