Week 6 H （A - 氪金带东）（B - 戴好口罩！）（C - 掌握魔法の东东 I）（D - 数据中心）

A - 氪金带东

题意

思路

首先，这个图是棵树
解法：三遍DFS
前两遍用来寻找树的直径,第一遍找到直径的一个端点，第二遍记录每个点到这个端点的距离并寻找直径的另一个端点，第三遍记录每个点到另外一个端点的距离。最后输出两者的最大值即可。

代码

#include<iostream>
#include<cstring>
#define llong long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
using namespace std;

const int maxn = 100005;

struct eg{
    int to, next;
    llong v;
}e[maxn];

llong n, tot, dia_len, dia_index;
llong head[maxn], vis[maxn], dis[maxn], tdis[maxn];

void add(int x, int y, llong v)
{
    e[++tot].to = y;
    e[tot].v = v;
    e[tot].next = head[x];
    head[x] = tot;
}

void dfs(int x)
{
    vis[x] = 1;
    if(dis[x]>=dia_len){
        dia_len = dis[x];
        dia_index = x;
    }
    for(int i = head[x];~i;i = e[i].next){
        int t = e[i].to;
        if(!vis[t]){
            vis[t] = 1;
            dis[t] = dis[x]+e[i].v;
            dfs(t);
        }
    }
}

void init()
{
    tot = 0; dia_len = 0; dia_index = 1;
    memset(head, -1, sizeof(head));
    memset(dis, 0, sizeof(dis));
    memset(vis,0,sizeof(vis));
}

int main(){
    ios::sync_with_stdio(false);
    while(cin>>n){
        init();
        For(i,2,n){
            int y;
            llong v;
            cin>>y>>v;
            add(i, y, v);
            add(y, i, v);
        }
        dfs(1);     
        memset(dis, 0, sizeof(dis));
        memset(vis,0,sizeof(vis));
        dfs(dia_index);
        For(i,1,n){
            tdis[i] = dis[i];
            dis[i] = 0;
            vis[i] = 0;
        }
        dfs(dia_index);
        For(i,1,n){
            cout<<max(dis[i],tdis[i])<<endl;
        }
    }

    return 0;
}

B - 戴好口罩！

题意

思路

并查集
好像没什么好说的

代码

#include<iostream>
#include<cstring>
#define llong long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
using namespace std;

const int maxn = 100005;

int n, m, ans;
int pre[maxn];
int find(int x){
    return pre[x] = (pre[x] == x ? x : find(pre[x]));
}

int main(){
    ios::sync_with_stdio(false);
    while(cin>>n>>m){
        ans = 1;
        if(!n && !m) break;
        For(i,0,n-1) pre[i] = i;
        For(i,1,m){
            int tot, fst;
            cin>>tot>>fst;
            For(i,1,tot-1){
                int num;
                cin>>num;
                if(pre[num] == num)
                    pre[num] = find(fst);
                else
                    pre[find(num)] = find(fst);
                fst = num;
            }
        }
        For(i,1,n-1){
            if(find(0)==find(i))
                ans++;
        }
        cout<<ans<<endl;
    }
    
    return 0;
}

C - 掌握魔法の东东 I

题意

思路

看似有向图其实无向图
最小生成树
这题需要添加一条从0号农田开始的边（处理“黄河之水天上来”）
用到Kruskal算法（每次选最短的边），用并查集来保证选的边不成环。
总共选择n条边（从0开始）

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#define llong long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
using namespace std;

const int maxn = 100005;

struct eg{
    int to, from;
    int v;
    bool operator < (const eg& e) const{
        return v<e.v;
    }
}e[maxn];

int n, tot, w, ans, cnt;
int pre[maxn];

void add(int x, int y, int v)
{
    e[++tot].from = x;
    e[tot].to = y;
    e[tot].v = v;
}

void init()
{
    tot = 0;
    For(i,1,n)
        pre[i] = i;
}

int find(int x) {return pre[x] = (pre[x]==x?x:find(pre[x]));}

void Kruskal()
{
    sort(e+1,e+tot+1);
    For(i,1,tot){
        int u = e[i].from;
        int v = e[i].to;
        int pu = find(u), pv = find(v);
        if(pu!=pv){
            cnt++;
            ans+=e[i].v;
            pre[pu] = pv;
        }
        if(cnt==n) break;
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n;
    init();
    For(i,1,n){
        cin>>w;
        add(0,i,w);
    }
    For(i,1,n){
        For(j,1,n){
            cin>>w;
            if(i!=j) add(i,j,w);
        }
    }
    Kruskal();
    cout<<ans<<endl;
    return 0;
}

D - 数据中心

题意

思路

还是一道最小生成树的题，思路和第三题基本相同。
这题和刚刚那题的不同之处在于：
C题是取和，这题是取最大值，也就是最小生成树中最长的边。

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#define llong long long
#define For(i,a,n) for(register int i=a;i<=n;i++)
using namespace std;

const int maxn = 200005;

struct eg{
    int to, from;
    int v;
    bool operator < (const eg& e) const{
        return v<e.v;
    }
}e[maxn];

int n, m, r, tot, u, v, w, ans, cnt;
int pre[maxn];

void add(int x, int y, int v)
{
    e[++tot].from = x;
    e[tot].to = y;
    e[tot].v = v;
}

void init()
{
    tot = 0;
    For(i,1,n)
        pre[i] = i;
}

int find(int x) {return pre[x] = (pre[x]==x?x:find(pre[x]));}

void Kruskal()
{
    sort(e+1,e+tot+1);
    For(i,1,tot){
        int u = e[i].from;
        int v = e[i].to;
        int pu = find(u), pv = find(v);
        if(pu!=pv){
            cnt++;
            ans = max(ans, e[i].v);
            pre[pu] = pv;
        }
        if(cnt==n-1) break;
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>r;
    init();
    For(i,1,m){
        cin>>u>>v>>w;
        add(u,v,w);
        add(v,u,w);
    }
    Kruskal();
    cout<<ans<<endl;
    return 0;
}