码迷,mamicode.com
首页 > 其他好文 > 详细

树形dp——cf1029E

时间:2019-05-05 01:29:55      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:back   end   code   题解   oid   void   color   div   c++   

题解给出的是带log的,,我自己写了个on的。。

#include<bits/stdc++.h>
#include<vector>
using namespace std;
#define maxn 200005
vector<int>G[maxn];

int d[maxn],ans,n,flag[maxn];
void dfs(int u,int pre,int dep){
    d[u]=dep;
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==pre)continue;
        dfs(v,u,dep+1);
    }
}
//三类结点:0没被访问过的,1:被连边的,2:距离<=2的 
void dfs2(int u,int pre){
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(v==pre)continue;
        dfs2(v,u);
    } 
    if(flag[u]==0 && d[u]>2)flag[pre]=1;    
    if(flag[u]==1){
        ans++;
        if(flag[pre]!=1)flag[pre]=2;
    }
}

int main(){
    cin>>n;
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    
    dfs(1,1,0);dfs2(1,1);
     
    cout<<ans<<endl;
} 

 

树形dp——cf1029E

标签:back   end   code   题解   oid   void   color   div   c++   

原文地址:https://www.cnblogs.com/zsben991126/p/10810544.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!