AtCoder ABC 157D Friend Suggestions

题目链接：https://atcoder.jp/contests/abc157/tasks/abc157_d

题目大意

　　有$N$个人，他们之间有$M$对直接好友关系和$K$对封闭关系。

　　求每个人有多少个间接好友。

分析

　　非常明显的并查集，不解释了。

代码如下

#include <bits/stdc++.h>
using namespace std;

/*-------------------Define Start-------------------*/
typedef bool BL;                        // 布尔类型
typedef char SB;                        // 有符号1字节,8位
typedef unsigned char UB;                // 无符号1字节,8位
typedef short SW;                        // 有符号短整型,16位
typedef unsigned short UW;                // 无符号短整型,16位
typedef long SDW;                        // 有符号整型,32位
typedef unsigned long UDW;               // 无符号整型,32位
typedef long long SLL;                    // 有符号长整型,64位
typedef unsigned long long ULL;            // 无符号长整型,64位
typedef char CH;                        // 单个字符
typedef float R32;                        // 单精度浮点数
typedef double R64;                        // 双精度浮点数

#define Rep(i, n) for (register SDW i = 0; i < (n); ++i)
#define For(i, s, t) for (register SDW i = (s); i <= (t); ++i)
#define rFor(i, t, s) for (register SDW i = (t); i >= (s); --i)
#define foreach(i, c) for (__typeof(c.begin()) i = c.begin(); i != c.end(); ++i)
#define ms0(a) memset(a,0,sizeof(a))
#define msI(a) memset(a,0x7f,sizeof(a))
#define LOWBIT(x) ((x)&(-x))

#define MP make_pair
#define PB push_back
#define ft first
#define sd second
#define ALL(x) x.begin(),x.end()

#define pr(x) cout << #x << " = " << x << "  "
#define prln(x) cout << #x << " = " << x << endl

const ULL mod = 1e9 + 7;                //常用模数(可根据题目需要修改)
const ULL inf = 0x7fffffff;                //用来表示无限大
const ULL infLL = 0x7fffffffffffffffLL;    //用来表示无限大
/*-------------------Define End-------------------*/

// 重载<<操作符，用来打印vector
template < typename T >
ostream& operator<< (ostream& out, vector< T > vec) {
    foreach(i, vec) {
        if(i != vec.begin()) {
            out << " ";
        }
        out << *i;
    }
    return out;
}

const UDW maxN = 1e5 + 7;
SDW N, M, K;
vector< SDW > ans;

SDW dsu[maxN], sz[maxN];    // DISJOINT_SET_UNION 并查集 

SDW Find(SDW x){
    while (x != dsu[x]) x = dsu[x] = dsu[dsu[x]];
    return x;
} 

void Union(SDW x, SDW y){
    if (sz[x] > sz[y]) swap(x, y);
    dsu[x] = y;
    sz[y] += sz[x];
}

void input(){
    cin >> N >> M >> K;
    
    // 并查集初始化 
    Rep(i, N) {
        dsu[i] = i;
        sz[i] = 1;
    }
    
    ans.assign(N, 0);
}

void solve(){
    SDW A, B, C, D;
    
    Rep(i, M) {
        cin >> A >> B;
        SDW x = Find(--A);
        SDW y = Find(--B);
        if(x != y) {
            Union(x, y);
        }
        --ans[A];    // 减去直接好友关系数 
        --ans[B];    // 减去直接好友关系数 
    }
    
    Rep(i, N) {
        SDW x = Find(i);
        ans[i] += sz[x] - 1; // 加上间接好友关系数和直接好友关系数 
    }
    
    Rep(i, K) {
        cin >> C >> D;
        SDW x = Find(--C);
        SDW y = Find(--D);
        if(x == y) {
            --ans[C];    // 减去间接坏友关系数 
            --ans[D];    // 减去间接坏友关系数 
        }
    }
}

void output(){
    cout << ans << endl;
}

int main() {
    input();
    solve();
    output();
    return 0;
}