/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 7ms 6.52 MiB
#2 Accepted 8ms 6.52 MiB
#3 Accepted 16ms 6.52 MiB
#4 Accepted 17ms 6.66 MiB
#5 Accepted 29ms 7.52 MiB
#6 Accepted 43ms 7.5 MiB
#7 Accepted 39ms 7.566 MiB
#8 Accepted 219ms 6.52 MiB
#9 Accepted 71ms 6.77 MiB
#10 Accepted 71ms 7.52 MiB
#11 Accepted 41ms 7.52 MiB
#12 Accepted 67ms 7.52 MiB
#13 Accepted 21ms 7.566 MiB
#14 Accepted 57ms 7.52 MiB
#15 Accepted 41ms 6.836 MiB
#16 Accepted 19ms 7.52 MiB
#17 Accepted 19ms 7.52 MiB
#18 Accepted 19ms 7.52 MiB
#19 Accepted 89ms 13.535 MiB
#20 Accepted 62ms 13.406 MiB
#21 Accepted 60ms 13.508 MiB
#22 Accepted 106ms 27.969 MiB
#23 Accepted 106ms 12.918 MiB
#24 Accepted 109ms 12.77 MiB
#25 Accepted 94ms 12.566 MiB
#26 Accepted 11ms 7.27 MiB

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 2e5 + 10;
const ll MOD = 1e9 + 7;

int root, cnt, mxlen;

vector<int> v(N), dist(N), g[N];

void dfs(int vertex, int par, int cur){
  dist[vertex] = cur;
  for (auto child : g[vertex]){
    if (child == par) continue;
    dfs(child, vertex, cur + 1);
    v[vertex] = max(v[vertex], v[child]);
  }
}

void dfs2(int vertex, int par, int cur) {
  mxlen = max(mxlen, cur);
  cnt++;
  for(auto child : g[vertex]) {
    if (child == par) continue;
    if(v[child]) {
      dfs2(child, vertex, cur + 1);
    }
  }
}

void solve()
{
  for(int i = 0; i < N; i++) {
    g[i].clear();
    dist[i] = 0;
    v[i] = 0;
    root = mxlen = cnt = 0;
  }
  int n;
  cin >> n;
  int root = 0;
  for(int i = 1; i <= n; i++) {
    cin >> v[i];
    if(v[i] == 1) {
      root = i;
    }
  }
  for(int i = 1; i < n; i++) {
    int u, v;
    cin >> u >> v;
    g[u].push_back(v);
    g[v].push_back(u);
  }

  dfs(root, 0, 0);

  int mx = 0;
  for(int i = 1; i <= n; i++) {
    if(v[i] and dist[i] > mx) {
      root = i;
      mx = dist[i];
    }
  }
  
  dfs2(root, 0, 0);
  int ans = (cnt - 1) * 2 - mxlen;

  cout << ans << endl;
}

int main()
{
  ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

  int t;
  cin >> t;
  while(t--){
    solve();
  }
}

Information

Submit By
Type
Submission
Problem
P1078 Apple on Tree
Language
C++20 (G++ 13.2.0)
Submit At
2024-08-17 20:21:08
Judged At
2024-08-17 20:21:08
Judged By
Score
100
Total Time
219ms
Peak Memory
27.969 MiB