/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 3ms 2.777 MiB
#2 Accepted 3ms 2.777 MiB
#3 Wrong Answer 2ms 3.277 MiB
#4 Wrong Answer 5ms 2.879 MiB
#5 Wrong Answer 27ms 3.777 MiB
#6 Wrong Answer 46ms 3.984 MiB
#7 Accepted 35ms 4.02 MiB
#8 Wrong Answer 21ms 2.977 MiB
#9 Wrong Answer 44ms 3.0 MiB
#10 Wrong Answer 56ms 4.02 MiB
#11 Wrong Answer 39ms 4.031 MiB
#12 Wrong Answer 46ms 3.953 MiB
#13 Wrong Answer 21ms 4.125 MiB
#14 Wrong Answer 60ms 4.121 MiB
#15 Wrong Answer 27ms 3.129 MiB
#16 Wrong Answer 19ms 4.027 MiB
#17 Accepted 19ms 3.949 MiB
#18 Accepted 17ms 4.566 MiB
#19 Wrong Answer 10ms 3.57 MiB
#20 Wrong Answer 9ms 3.527 MiB
#21 Wrong Answer 8ms 3.574 MiB
#22 Wrong Answer 8ms 3.57 MiB
#23 Wrong Answer 8ms 3.527 MiB
#24 Wrong Answer 9ms 3.527 MiB
#25 Wrong Answer 9ms 3.527 MiB
#26 Wrong Answer 7ms 4.055 MiB

Code

#include <bits/stdc++.h>
using namespace std;
#define SC               scanf
#define PF               printf
#define ull              unsigned long long
#define ld               long double
#define F                first
#define S                second
#define pb               push_back
#define sort_a(a)        sort(a.begin(),a.end());
#define sort_d(a)        sort(a.rbegin(),a.rend());
#define READ(f)          freopen(f, "r", stdin)
#define WRITE(f)         freopen(f, "w", stdout)
#define rev(s)           reverse(s.begin(),s.end())
#define P(ok)            cout << (ok ? "YES\n": "NO\n")
#define __Heart__              ios_base :: sync_with_stdio(false); cin.tie(NULL);
#define ll long long
typedef pair< ll , ll>                   PII;
const int MX = 1e5 + 5 ;
vector<int> Edge[MX] ;
int cnt = 0  , n , u , v , farthestNode = 0 , days = 0 , lev[MX] , A[MX];
void dfs(int id , int par){
  for(auto child : Edge[id]){
    if(child != par){
        lev[child] = lev[id] + 1 ;
        if(days < lev[child] ){
            days = lev[child] ;
           if(A[child]) farthestNode = child ;
        }
        dfs(child , id) ;
    }
  }
}
void dfs2(int id , int par){
  for(auto child : Edge[id]){
    if(child != par){
        lev[child] = lev[id] + 1 ;
        dfs2(child , id) ;
    }
  }
}
void solve()
{
    cin >> n ;
    int apple = 0 , st = 0 , en = 0;
    days = 0 , farthestNode = 0;
    for(int i = 1 ; i <= n ; i++) {cin >> A[i] , lev[i] = 0 , Edge[i].clear(); if(A[i]) apple = i ;}
    for(int i = 1 ; i < n ; i++){
        cin >> u >> v ;
        Edge[u].pb(v) ;
        Edge[v].pb(u) ;

    }
    dfs(apple , -1) ;
    for(int i = 1 ; i <= n ; i++) lev[i] = 0;
    cnt = 0 , days = 0 , st = farthestNode;
    dfs(farthestNode , -1) ;
    en = farthestNode ;
    for(int i = 1 ; i <= n ; i++) lev[i] = 0 ;
    dfs2(st , -1) ;
    map < int , int > freq ;
    for(int i = 1 ; i <= n ; i++)
    {
        if(A[i]) freq[lev[i]]++ ;
    }
    ll Ans = 0 ;
    for(int i = 1 ; i <= lev[en] ; i++){
        if(freq[i] > 1) {
            Ans += ((freq[i] - 1) * 2 );
        }
        Ans ++ ;
    }
    if(!en) Ans = 0 ;
    cout << Ans << "\n";

}
int main()
{
     __Heart__
     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-03 02:26:38
Judged At
2024-08-03 02:26:38
Judged By
Score
13
Total Time
60ms
Peak Memory
4.566 MiB