蓝桥杯-网络分析

网络分析

小明正在做一个网络实验。

他设置了 n 台电脑，称为节点，用于收发和存储数据。

初始时，所有节点都是独立的，不存在任何连接。

小明可以通过网线将两个节点连接起来，连接后两个节点就可以互相通信了。

两个节点如果存在网线连接，称为相邻。

小明有时会测试当时的网络，他会在某个节点发送一条信息，信息会发送到每个相邻的节点，之后这些节点又会转发到自己相邻的节点，直到所有直接或间接相邻的节点都收到了信息。

所有发送和接收的节点都会将信息存储下来。

一条信息只存储一次。

给出小明连接和测试的过程，请计算出每个节点存储信息的大小。

输入格式
输入的第一行包含两个整数 n,m，分别表示节点数量和操作数量。

节点从 1 至 n 编号。

接下来 m 行，每行三个整数，表示一个操作。

如果操作为 1 a b，表示将节点 a 和节点 b 通过网线连接起来。当 a = b 时，表示连接了一个自环，对网络没有实质影响。
如果操作为 2 p t，表示在节点 p 上发送一条大小为 t 的信息。
输出格式
输出一行，包含 n 个整数，相邻整数之间用一个空格分割，依次表示进行完上述操作后节点 1 至节点 n 上存储信息的大小。

数据范围
\(1≤n≤10000,\)
\(1≤m≤105,\)
\(1≤t≤100\)
输入样例1：

4 8
1 1 2
2 1 10
2 3 5
1 4 1
2 2 2
1 1 2
1 2 4
2 2 1

输出样例1：

13 13 5 3

dfs暴力做法

该方法可以拿70%的分数，适合骗分

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i = a;i < b;i++)
#define per(i,a,b) for(int i = b - 1;i >= a;i--)
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
typedef long long ll;
typedef pair<int,int> PII;
typedef vector<int> VI;

const int N = 10010;

int cnt[N], n, m;
bool vis[N];
VI G[N];

void dfs(int u, int t) {
    vis[u] = 1;
    cnt[u] += t;
    for(auto v : G[u]) {
        if(vis[v]) continue;
        dfs(v,t);
    }
}

int main() {
    scanf("%d%d", &n, &m);
    rep(_,0,m) {
        int op;
        scanf("%d", &op);
        if(op ==  1) {
            int a, b;
            scanf("%d%d", &a, &b);
            G[a].push_back(b);
            G[b].push_back(a);
        }else {
            int p, t;
            scanf("%d%d", &p, &t);
            memset(vis,0,sizeof(vis));
            dfs(p,t);
        }
    }
    rep(i,1,n+1) printf("%d ", cnt[i]);
    puts("");
    return 0;
}

正解：并查集的高级操作

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i = a;i < b;i++)
#define per(i,a,b) for(int i = b - 1;i >= a;i--)
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
typedef long long ll;
typedef pair<int,int> PII;
typedef vector<int> VI;

const int N = 10010;

int fa[N], d[N], n, m;
int find(int x) {
    if(x == fa[x] || fa[fa[x]] == fa[x]) return fa[x];
    int t = find(fa[x]);
    d[x] += d[fa[x]];
    fa[x] = t;
    return fa[x];
}


int main() {
    scanf("%d%d", &n, &m);
    rep(i,1,n+1) fa[i] = i;
    rep(_,0,m) {
        int op; 
        scanf("%d", &op);
        if(op ==  1) {
            int a, b;
            scanf("%d%d", &a, &b);
            int fx = find(a), fy = find(b);
            if(fx != fy) {
                d[fx] -= d[fy];
                fa[fx] = fy;
            }
        }else {
            int p, t;
            scanf("%d%d", &p, &t);
            d[find(p)] += t;
        }
    }
    rep(i,1,n+1) {
        if(i == find(i)) printf("%d ", d[i]);
        else printf("%d ", d[i] + d[find(i)]);
    }
    puts("");
    return 0;
}