結果

問題 No.1153 ねこちゃんゲーム
ユーザー ryoissy
提出日時 2020-08-15 20:23:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 488 ms / 2,500 ms
コード長 1,661 bytes
コンパイル時間 2,007 ms
コンパイル使用メモリ 183,312 KB
実行使用メモリ 86,784 KB
最終ジャッジ日時 2024-10-10 22:38:32
合計ジャッジ時間 23,106 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define MOD 1000000007LL
using namespace std;
typedef long long ll;
typedef pair<int,int> P;

int n,m;
int a[200005];
int dp[200005];
int dp2[200005];
vector<int> G[200005];
vector<P> G2[200005];
bool checked[200005];

int dfs(int v,int p){
	set<int> se;
	for(int i=0;i<G[v].size();i++){
		int nv=G[v][i];
		if(nv!=p){
			se.insert(dfs(nv,v));
		}
	}
	int res=0;
	while(se.find(res)!=se.end()){
		res++;
	}
	return (dp[v]=res);
}

void dfs2(int v,int p,int val){
	//printf("dfs2 %d %d %d\n",v,p,val);
	map<int,int> se;
	for(int i=0;i<G[v].size();i++){
		int nv=G[v][i];
		if(nv!=p){
			se[dp[nv]]++;
			G2[v].push_back(P(nv,dp[nv]));
		}
	}
	se[val]++;
	int res=0;
	while(se[res]!=0){
		res++;
	}
	dp2[v]=res;
	for(int i=0;i<G[v].size();i++){
		int nv=G[v][i];
		if(nv!=p){
			if(dp[nv]<res && se[dp[nv]]==1){
				dfs2(nv,v,dp[nv]);
				G2[nv].push_back(P(v,dp[nv]));
			}else{
				dfs2(nv,v,res);
				G2[nv].push_back(P(v,res));
			}
		}
	}
}

int main(void){
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d",&a[i]);
		a[i]--;
	}
	for(int i=0;i<n-1;i++){
		int f,t;
		scanf("%d%d",&f,&t);
		f--;
		t--;
		G[f].push_back(t);
		G[t].push_back(f);
	}
	dfs(0,-1);
	dfs2(0,-1,-1);
	for(int i=0;i<n;i++){
		//printf("%d v %d %d\n",i,dp[i],dp2[i]);
	}
	int xsum=0;
	for(int i=0;i<m;i++){
		xsum^=dp2[a[i]];
	}
	if(xsum==0){
		printf("-1 -1\n");
		return 0;
	}
	for(int i=0;i<m;i++){
		if(checked[a[i]])continue;
		for(int j=0;j<G2[a[i]].size();j++){
			int nv=G2[a[i]][j].first;
			int val=G2[a[i]][j].second;
			if((dp2[a[i]]^val)==xsum){
				printf("%d %d\n",i+1,nv+1);
				return 0;
			}
		}
		checked[a[i]]=true;
	}
	return 0;
}
0