Submission #377830

#	Time	Username	Problem	Language	Result	Execution time	Memory
377830		autumn_eel	기지국 (IOI20_stations)	C++14	10 / 100	986 ms	1060 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
#define rep(i,n)for(int i=0;i<int(n);i++)
using namespace std;
typedef long long ll;

static vector<vector<int>>E;
static vector<int>labels;
static int cnt=0;

void dfs(int v,int p){
	int x=0;
	int d=0;
	for(int u:E[v]){
		if(u==p)continue;
		dfs(u,v);
		x|=labels[u];
		d++;
	}
	if(d<=1){
		x|=1<<cnt;
		cnt++;
	}
	labels[v]=x;
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v){
	E=vector<vector<int>>(n);
	labels=vector<int>(n);
	cnt=0;
	vector<int>deg(n);
	rep(i,n-1){
		E[u[i]].push_back(v[i]);
		E[v[i]].push_back(u[i]);
		deg[u[i]]++;
		deg[v[i]]++;
	}
	dfs(0,-1);
	return labels;
}

int find_next_station(int s, int t, std::vector<int> c) {
	for(int u:c){
		if((s|u)==u)continue;
		if((t|u)==u)return u;
	}
	for(int u:c){
		if((s|u)==u)return u;
	}
	return -1;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 8
• Subtask 3: 0 / 16
• Subtask 4: 10 / 10
• Subtask 5: 0 / 61