美文网首页动态规划
Codeforces Round #506 (Div. 3)(T

Codeforces Round #506 (Div. 3)(T

作者: kimoyami | 来源:发表于2018-08-27 22:10 被阅读15次

    链接:https://codeforces.com/contest/1029/problem/E
    思路:一开始想用树形dp做,但后来怎么做都有错,想法时用dp(u,f,flag)表示状态,u表示当前节点,f表示父节点,flag表示上一个节点是否与1连通.先做一遍dfs把与1节点距离在2以内的点全部先标记出来,然后如果flag为0或者与1距离位2及以内则不需要+1,否则必须连边+1,然后直到无路可走最后就是正确答案,但是不知道哪里错了= =,就有几个数据过不了不知道是哪种特殊情况没有考虑到QAQ。然后说一下官方的做法吧,直接遍历一遍,因为连通且只有n-1条边,所以到每个点的路有且仅有一条,所以距离也是唯一确定的,直接一遍dfs更新所有点的距离,然后把距离大于2的点放入一个set中,然后我们考虑,因为是有向图且只有一条路径,所以如果你连接一个点,他有可能为根节点,结果一定没有连他的父节点优,那父节点是不是一定是最优的呢,答案是肯定的,因为父节点一定要连,不然根节点就没法在距离2以内,所以我们就贪心,遍历整个set,然后找到他的父节点,连边并将+1,然后将这个父节点指向的所有其他节点(如果在set集合内)全部从set中删去,直到最终set只有一个节点,最后这个节点需要特判,直接+1即可= =,这样的贪心一定得到的是最优解
    代码:

    #include<bits/stdc++.h>
    using namespace std;
    
    int n;
    const int maxn = 2e5+100;
    vector<int> G[maxn];
    //这一部分是dp代码错误的,不知道哪里错了放在这里吧- -
    /*int used[maxn];
    int d[maxn];
    int ans = 0;
    
    void dfs(int u,int f){
    bool flag = false;
        for(int i=0;i<G[u].size();i++){
            int v = G[u][i];
            if(v!=f){
                d[v] = d[u] + 1;
                dfs(v,u);
                flag|=used[v];
            }
        }
        if(d[u]>2&&!used[u]&&!flag&&!used[f]){
            used[f] = true;
            ans++;
        }
    }
    
    int main(){
        scanf("%d",&n);
        for(int i=0;i<n-1;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(1,0);
        printf("%d\n",ans);
        return 0;
    }*/
    
    int d[maxn];
    int f[maxn];
    set<pair<int,int> > jj;
    
    void dfs(int u,int p){
        f[u] = p;
        for(int i=0;i<G[u].size();i++){
            int v = G[u][i];
            if(v!=p){
                d[v] = d[u] + 1;
                dfs(v,u);   
            }   
        }
    }
    
    int main(){
        scanf("%d",&n);
        for(int i=0;i<n-1;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(1,-1);
        for(int i=1;i<=n;i++){
            if(d[i]>2)jj.insert(make_pair(-d[i],i));
        }   
        int ans = 0;
        while(!jj.empty()){
            int v = jj.begin()->second;
            v = f[v];
            ++ans;
            auto it = jj.find(make_pair(-d[v],v));//可以学习一下auto的用法
            if(it!=jj.end())jj.erase(it);
            for(auto to : G[v]){
                auto it = jj.find(make_pair(-d[to],to));
                if(it!=jj.end())
                              jj.erase(it);从set中剔除it元素
            }
        }
        printf("%d\n",ans);
        return 0;
    }
    

    相关文章

      网友评论

        本文标题:Codeforces Round #506 (Div. 3)(T

        本文链接:https://www.haomeiwen.com/subject/qxhxwftx.html