/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 8ms 6.52 MiB
#2 Accepted 7ms 6.52 MiB
#3 Accepted 11ms 6.52 MiB
#4 Accepted 11ms 6.52 MiB
#5 Accepted 31ms 7.52 MiB
#6 Accepted 71ms 7.52 MiB
#7 Accepted 37ms 7.523 MiB
#8 Accepted 211ms 6.668 MiB
#9 Accepted 88ms 6.77 MiB
#10 Accepted 56ms 7.523 MiB
#11 Accepted 42ms 7.52 MiB
#12 Accepted 48ms 7.535 MiB
#13 Accepted 23ms 7.52 MiB
#14 Accepted 63ms 7.52 MiB
#15 Accepted 42ms 6.848 MiB
#16 Accepted 20ms 7.473 MiB
#17 Accepted 20ms 7.516 MiB
#18 Accepted 20ms 7.566 MiB
#19 Accepted 62ms 13.383 MiB
#20 Accepted 84ms 13.551 MiB
#21 Accepted 63ms 13.543 MiB
#22 Accepted 83ms 24.77 MiB
#23 Accepted 123ms 12.77 MiB
#24 Accepted 119ms 12.918 MiB
#25 Accepted 75ms 12.77 MiB
#26 Accepted 10ms 7.039 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){
  for (auto child : g[vertex]){
    if (child == par) continue;
    dfs(child, vertex);
    v[vertex] = max(v[vertex], v[child]);
  }
}
void dfs2(int vertex, int par, int cur) {
  mxlen = max(mxlen, cur);
  cnt++;
  dist[vertex] = cur;
  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);
  dfs2(root, 0, 0);

  int ans = (cnt - 1) * 2, mx = 0;
  for(int i = 1; i <= n; i++) {
    if(dist[i] > mx) {
      root = i;
      mx = dist[i];
    }
  }
  dfs2(root, 0, 0);
  ans -= 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++17 (G++ 13.2.0)
Submit At
2024-08-17 20:09:03
Judged At
2024-08-17 20:09:03
Judged By
Score
100
Total Time
211ms
Peak Memory
24.77 MiB