/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 13ms 32.027 MiB
#2 Wrong Answer 12ms 31.543 MiB
#3 Wrong Answer 14ms 32.559 MiB
#4 Wrong Answer 14ms 32.57 MiB
#5 Wrong Answer 36ms 32.77 MiB
#6 Wrong Answer 55ms 32.902 MiB
#7 Wrong Answer 44ms 34.27 MiB
#8 Wrong Answer 28ms 31.559 MiB
#9 Wrong Answer 52ms 32.52 MiB
#10 Wrong Answer 64ms 33.316 MiB
#11 Wrong Answer 51ms 34.328 MiB
#12 Wrong Answer 56ms 32.777 MiB
#13 Wrong Answer 31ms 36.242 MiB
#14 Wrong Answer 69ms 32.77 MiB
#15 Wrong Answer 36ms 32.52 MiB
#16 Wrong Answer 28ms 32.52 MiB
#17 Wrong Answer 29ms 36.27 MiB
#18 Wrong Answer 29ms 33.602 MiB
#19 Wrong Answer 72ms 44.133 MiB
#20 Wrong Answer 72ms 39.785 MiB
#21 Wrong Answer 72ms 44.934 MiB
#22 Wrong Answer 97ms 52.77 MiB
#23 Wrong Answer 135ms 39.555 MiB
#24 Wrong Answer 144ms 46.02 MiB
#25 Wrong Answer 86ms 39.27 MiB
#26 Wrong Answer 18ms 35.48 MiB

Code

#include <bits/stdc++.h>
#define ll long long
#define endll '\n';
#define pb push_back
#define all(v) v.begin(), v.end()
using namespace std;

const int mod = 1e9 + 7, N = 1e6;
int n;
vector<int> g[N + 3];

int mx1, mx2;
int ar[N + 3], vis[N + 3];
int dis[N + 3];
int lastNode;
int tracked[N + 3];
int par[N + 3];
int cost[N + 3];
void dfs1(int u)
{
  vis[u] = 1;

  for (auto v : g[u])
  {
    if (vis[v] == 0)
    {
      dis[v] = dis[u] + 1;
      dfs1(v);
    }
  }
}

void dfs2(int u)
{
  vis[u] = 1;

  for (auto v : g[u])
  {
    if (vis[v] == 0)
    {
      par[v] = u;
      dfs2(v);
    }
  }
}

void dfs3(int u)
{
  vis[u] = 1;

  for (auto v : g[u])
  {
    if (vis[v] == 0)
    {

      dfs3(v);
    }
  }

  if (tracked[u] == 0)
  {
    cost[par[u]] += cost[u] * 2 + ar[u] * 2;
  }
  else
  {
    if (cost[u])
    {
      cost[par[u]] += cost[u] + 1;
    }
    cost[par[u]] += ar[u];
  }

  // cout << u << "__ " << cost[par[u]] << endll;
}

void solve()
{
  cin >> n;
  for (int i = 1; i <= n; i++)
  {
    cin >> ar[i];
  }

  for (int i = 1; i < n; i++)
  {
    int u, v;
    cin >> u >> v;
    g[u].pb(v);
    g[v].pb(u);
  }

  dfs1(1);

  mx1 = 1e9, mx2 = -1;
  int node1 = -1, node2;
  for (int i = 1; i <= n; i++)
  {

    vis[i] = 0;
    if (ar[i])
    {
      // cout << "dukse = " << i << endll;
      if (mx1 > dis[i])
      {
        mx1 = dis[i];
        node1 = i;
      }

      if (mx2 < dis[i])
      {
        mx2 = dis[i];
        node2 = i;
      }
    }
  }

  if(node1 == -1) {
    cout << 0 << endll;
    return;
  }
  dfs2(node1);

  // cout << node1 << " " << node2 << endl;
  while (node2 != node1)
  {
    tracked[node2] = 1;
    node2 = par[node2];
  }
  tracked[node1] = 1;
  for (int i = 1; i <= n; i++)
  {
    // cout << i << "_" << tracked[i] << endl;
  }

  for (int i = 1; i <= n; i++)
    vis[i] = 0;

  dfs3(node1);

  cout << cost[node1] << endll;
  for (int i = 1; i <= n; i++)
  {
    g[i].clear();
    vis[i] = par[i] = cost[i] = ar[i] = dis[i] = lastNode = tracked[i] =0;
        mx1 = -1,
    mx2 = 0;
  }
}

int32_t main()
{
  ios::sync_with_stdio(false);
  cin.tie(0);
  int t = 1;
  cin >> t;
  while (t--)
  {
    solve();
  }
  return 0;
}

Information

Submit By
Type
Submission
Problem
P1078 Apple on Tree
Language
C++20 (G++ 13.2.0)
Submit At
2024-08-16 19:32:51
Judged At
2024-08-16 19:32:51
Judged By
Score
1
Total Time
144ms
Peak Memory
52.77 MiB