/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 4ms 2.812 MiB
#2 Accepted 71ms 7.836 MiB
#3 Accepted 67ms 7.789 MiB
#4 Accepted 77ms 7.918 MiB
#5 Accepted 76ms 7.914 MiB
#6 Accepted 76ms 7.789 MiB
#7 Accepted 76ms 7.918 MiB
#8 Accepted 72ms 7.91 MiB
#9 Accepted 83ms 7.91 MiB
#10 Accepted 77ms 7.906 MiB
#11 Accepted 76ms 7.918 MiB
#12 Accepted 76ms 7.793 MiB
#13 Accepted 77ms 7.918 MiB
#14 Accepted 76ms 7.918 MiB
#15 Accepted 77ms 7.918 MiB
#16 Accepted 77ms 7.91 MiB
#17 Accepted 67ms 7.812 MiB
#18 Accepted 64ms 7.684 MiB
#19 Accepted 75ms 7.914 MiB
#20 Accepted 80ms 7.793 MiB
#21 Accepted 5ms 2.715 MiB
#22 Accepted 5ms 2.82 MiB
#23 Accepted 49ms 16.789 MiB
#24 Accepted 50ms 16.801 MiB
#25 Accepted 4ms 2.812 MiB
#26 Accepted 23ms 3.992 MiB

Code

#include<bits/stdc++.h>
using namespace std;
const long long M=1e5+10,MOD=1000000007;
typedef long long ll;
vector<int>e[M];
int level[M];
int lv[M];
int ans=1;
int target_distane=0;// using method re-rooting dp
int dp[M][2];
int vis[M];
vector<int>v;
void dfs1(int x,int p){
    for(int u:e[x]){
        if(u!=p){
            int cur_level=dp[x][0];
            if(dp[x][0]==level[u]+1)cur_level=dp[x][1];
            cur_level=max(cur_level,lv[x]);
            int total=cur_level+level[u];
            total/=2;
            lv[u]=cur_level+1;// root to down length
            if(total>=target_distane){
                if(total==target_distane){
                    if(cur_level==level[u])ans=max(ans,max(u,x));
                    else{
                        int dif=cur_level-level[u];
                        if(dif==1)ans=max(ans,x);
                        if(dif==-1)ans=max(ans,u);
                    }
                }
                else{
                    if(cur_level==level[u]){
                        ans=max(u,x);
                        target_distane=total;
                    }
                    else{
                        int dif=cur_level-level[u];
                        if(dif==1)ans=x,target_distane=total;
                        if(dif==-1)ans=u,target_distane=total;
                    }
            }
        }

            dfs1(u,x);
        }
    }
}
void dfs(int x,int p){
    dp[x][1]=dp[x][0]=1;
	vis[x]=1;
    for(int u:e[x]){
        if(vis[u]==0){
			vis[u]=1;
            dfs(u,x);
        }
    }
    if(p!=-1)
    level[p]=max(level[p],level[x]+1);
    if(p!=-1){
        int cur=level[x]+1;
      if(cur>=dp[p][0]){
		  dp[p][1]=dp[p][0];
		  dp[p][0]=cur;
	  }
		else{
			dp[p][1]=max(dp[p][1],cur);
		}
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t=1;
    //cin>>t;
    while(t--){
     int n;
     cin>>n;
     for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        e[u].push_back(v);
        e[v].push_back(u);
     }
     for(int i=1;i<=n;i++){
        level[i]=1;
		 dp[i][0]=0;
		 dp[i][1]=0;
     }
     dfs(1,-1);// lower level and upper level length
     dfs1(1,0);
     cout<<target_distane<<" "<<ans<<"\n";
     
     
    }
    
   
   return 0;
    
}
 

Information

Submit By
Type
Submission
Problem
P1069 Vaccination
Language
C++11 (G++ 13.2.0)
Submit At
2024-07-11 17:55:21
Judged At
2024-10-03 13:41:30
Judged By
Score
100
Total Time
83ms
Peak Memory
16.801 MiB