#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <string>
#include <utility>
#include <vector>
using namespace std;
long n, minM = 1000000000, maxP = 0, mag[1000000], pd[1000000] = {}, path[1000000] = {}, pn[1000000] = {};
vector <long> adj[1000000], mi[1000000];
void dfs(int i, int p) {
minM = min(minM, mag[i]);
for (int a : adj[i]) {
if (a != p) {
dfs(a, i);
}
}
if (mag[i] == 1) {
pd[i] = 1;
for (int a : adj[i]) {
if (a != p && mag[a] == 1) {
pd[i] = max(pd[i], pd[a]+1);
}
}
for (int a : adj[i]) {
if (a != p && mag[a] == 1) {
if (pd[a]+1 == pd[i]) {
mi[i].push_back(a);
}
}
}
path[i] = pd[i];
if (mi[i].size() == 1){
long mind = mi[i][0];
for (int a : adj[i]){
if (a != p && mag[a] == 1 && a != mind) {
path[i] = min(path[i], pd[i]+pd[a]);
}
}
for (int a : adj[i]){
if (a != p && mag[a] == 1 && a != mind) {
if (pd[i]+pd[a] == path[i]){
mi[i].push_back(a);
}
}
}
} else if (mi[i].size() > 1){
path[i] = 2*pd[i]-1;
}
maxP = max(path[i], maxP);
}
}
void dfs2(int i, int v){
pn[i] = v;
for (int a : mi[i]) {
dfs2(a, v);
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n;
for (int i = 0; i < n-1; i++) {
int a, b;
cin >> a >> b;
adj[a-1].push_back(b-1);
adj[b-1].push_back(a-1);
}
for (int i = 0; i < n; i++) {
cin >> mag[i];
}
dfs(0, -1);
if (minM == 1){
for (int i = 0; i < n; i++){
if (path[i] == maxP){
dfs2(i, i+1);
}
}
bool twok = false;
for (int i = 0; i < n; i++){
if (mag[i] == 2){
int count = 0;
for (int a : adj[i]){
if (pn[a] != 0){
count++;
if (count >= 2){
twok = true;
break;
}
}
}
}
}
if (twok){
cout << "2/" << (2*maxP+1);
} else {
cout << "1/" << maxP;
}
} else {
cout << minM << "/1";
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
33 ms |
47360 KB |
Output is correct |
2 |
Correct |
32 ms |
47360 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
33 ms |
47452 KB |
Output is correct |
2 |
Correct |
33 ms |
47608 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
479 ms |
173944 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
32 ms |
47356 KB |
Output is correct |
2 |
Correct |
511 ms |
217752 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
610 ms |
247416 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
503 ms |
155768 KB |
Output is correct |
2 |
Correct |
383 ms |
127228 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
428 ms |
129244 KB |
Output is correct |
2 |
Incorrect |
108 ms |
56772 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
113 ms |
56440 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
443 ms |
135740 KB |
Output is correct |
2 |
Correct |
504 ms |
147192 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
504 ms |
152952 KB |
Output is correct |
2 |
Incorrect |
444 ms |
94072 KB |
Output isn't correct |