#include "catdog.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> adj[100001];
bool ss = 0;
void initialize(int N,vector<int> A,vector<int> B){
if(N==1)ss = 1;
for(int i = 0;i<N-1;i++){
adj[A[i]].push_back(B[i]);
adj[B[i]].push_back(A[i]);
}
}
int R[100001],B[100001];
int col[100001];
void solve(int i,int pr){
R[i] = 1e9;B[i] = 1e9;
for(auto j:adj[i]){
if(j==pr)continue;
solve(j,i);
R[i] = min(R[j],B[j]+1);
B[i] = min(R[i]+1,B[i]);
}
if(adj[i].size()==1&&pr!=0){
R[i] = 0 , B[i] = 0;
}
if(col[i]==1){
B[i] = 1e9;
}if(col[i]==2){
R[i] = 1e9;
}
}
int cat(int v){
if(ss)return 0;
col[v] = 1;
solve(1,0);
return min(R[1],B[1]);
}int dog(int v){
if(ss)return 0;
col[v] = 2;
solve(1,0);
return min(R[1],B[1]);
}int neighbor(int v){
if(ss)return 0;
col[v] = 0;
solve(1,0);
return min(R[1],B[1]);
}/*
int main(){
int N = 5;int A[] = {1,2,2,4};
int BB[] = {2,3,4,5};
initialize(N,A,BB);
cout<<cat(3)<<endl;
cout<<dog(5)<<endl;
}
*/
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
2644 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |