Problem p02762 (C++)

Buggy submission (status: Runtime Error):
#include <algorithm>
#include <cmath>
#include <fstream>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <vector>

using namespace std;

#define I_MAX 2147483647
#define LL_MAX 9223372036854775807
#define ll long long
#define ld long double

struct XX {
  string s;
  int p;
  int c1;
  int c2;
  int num;
};

class xxGreater {
public:
  bool operator()(const XX &riLeft, const XX &riRight) const {
    // 第2条件
    if ((riLeft.s) == (riRight.s)) {
      return riLeft.p <
             riRight.p; //<:昇順(小さいものから順番)、>:降順(大きいものから順番)
      // プライオリティキューの場合は > で、top()すると値の小さいものがとれる
    }
    // 第1条件
    return (riLeft.s) < (riRight.s);
  }
};

// map<long long,long long> prime_f(long long n){
//     map<long long,long long>res;
//     for(int i=2;i*i<=n;i++){
//         while(n%i==0){
//             ++res[i];
//             n/=i;
//         }
//     }
//     if(n!=1)res[n]=1;
//     return res;
// }

// int n;
////int dat[2*10000000];
////int dat2[2*10000000];
// int dat[10];
// int dat2[10];
//
// void init(int n_){
//     n=1;
//     while(n<n_)n*=2;
//     for(int i=0;i<2*n-1;i++){
//         dat[i]=0;
//         dat2[i]=0;
//     }
// }
//
// void initset(int k,int a){
//     k+=n-1;
//     dat[k]=a;
//     while(k>0){
//         k=(k-1)/2;
//         dat[k]=dat[k*2+1]+dat[k*2+2];
//     }
// }
//
////[a,b)の間を[l,r]区間で比較しアップデート
////引数のindexに注意
////nは固定。initで計算すみ
////update2(L[i],R[i]+1,0,0,n,D[i]);
// void update2(int a,int b,int k,int l,int r,int v){//v更新値、区間は0-index
//     if(r<=a || b<=l)return;
//     if(a<=l && r<=b){
//         dat[k]+=dat2[k];
//         if(r-l>1){
//             dat2[k*2+1]+=dat2[k]/2;
//             dat2[k*2+1]+=dat2[k]/2;
//         }
//         dat2[k]=v*(r-l);
//         return;
//     }else{
//         update2(a,b,k*2+1,l,(l+r)/2,v);
//         update2(a,b,k*2+2,(l+r)/2,r,v);
//         return;
//     }
// }
//
// int query(int a,int b,int k,int l,int r){
//     if(r<=a || b<=l)return 0;
//     if(a<=l && r<=b){
//         dat[k]+=dat2[k];
//         if(r-l>1){
//             dat2[k*2+1]+=dat2[k]/2;
//             dat2[k*2+1]+=dat2[k]/2;
//         }
//         dat2[k]=0;
//         return dat[k];
//     }
//     else{
//         int vl=query(a,b,k*2+1,l,(l+r)/2);
//         int vr=query(a,b,k*2+2,(l+r)/2,r);
//         return vl+vr;
//     }
// }

// void printb(unsigned int v) {
//     unsigned int mask = (int)1 << (sizeof(v) * CHAR_BIT - 1);
//     do putchar(mask & v ? '1' : '0');
//     while (mask >>= 1);
// }

#ifdef DEBUG

#else

#endif
set<int> G1[10001];
set<int> G2[10001];
ll oya[10001];
ll nak[10001];
ll non[10001];

int ppar[100001];  // 親(1-index)
int rrank[100001]; // 木の深さ

void init(int n) { //
  for (int i = 1; i <= n; i++) {
    ppar[i] = i;
    rrank[i] = 0;
  }
}

int find(int x) {
  if (ppar[x] == x) {
    return x;
  } else {
    return ppar[x] = find(ppar[x]);
  }
}

void unite(int x, int y) {
  x = find(x);
  y = find(y);
  if (x == y) {
    return;
  }
  if (rrank[x] < rrank[y]) {
    ppar[x] = y;
  } else {
    ppar[y] = x;
    if (rrank[x] == rrank[y]) {
      rrank[x]++;
    }
  }
}
bool same(int x, int y) { return find(x) == find(y); }

int main(int argc, const char *argv[]) {
  // scanf("%s",S);
  // scanf("%d",&N);
  // scanf("%lld %lld",&target1,&target2);
  // sscanf(tmp.c_str(),"%dd%d%d",&time[i], &dice[i], &z[i]);
  // getline(cin, target);
  // ifstream ifs("01");//テスト用
  // ifs >> a;
  // ここから

  // 入力高速化
  ios::sync_with_stdio(false);
  cin.tie(0);

  int N, M, K;
  cin >> N >> M >> K;
  init(N);
  for (int i = 0; i < M; i++) {
    int t1, t2;
    cin >> t1 >> t2;
    G1[t1].insert(t2);
    G1[t2].insert(t1);
    unite(t1, t2);
  }
  for (int i = 0; i < K; i++) {
    int t1, t2;
    cin >> t1 >> t2;
    if (same(t1, t2)) {
      G2[t1].insert(t2);
      G2[t2].insert(t1);
    }
  }
  for (int i = 1; i <= N; i++) {
    int t = find(i);
    oya[t]++;
  }

  for (int i = 1; i <= N; i++) {
    int ans = 0;
    int t = find(i);
    ans += oya[t] - 1;
    ans -= G1[i].size();
    ans -= G2[i].size();
    cout << ans << " ";
  }
  cout << endl;

  // ここまで
  // cout << "ans" << endl;
  // cout << " " << "ans" << endl;
  // printf("%.0f\n",ans);//小数点以下表示なし
  // printf("%.7f\n",p);

  return 0;
}


Fixed submission:
#include <algorithm>
#include <cmath>
#include <fstream>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <vector>

using namespace std;

#define I_MAX 2147483647
#define LL_MAX 9223372036854775807
#define ll long long
#define ld long double

struct XX {
  string s;
  int p;
  int c1;
  int c2;
  int num;
};

class xxGreater {
public:
  bool operator()(const XX &riLeft, const XX &riRight) const {
    // 第2条件
    if ((riLeft.s) == (riRight.s)) {
      return riLeft.p <
             riRight.p; //<:昇順(小さいものから順番)、>:降順(大きいものから順番)
      // プライオリティキューの場合は > で、top()すると値の小さいものがとれる
    }
    // 第1条件
    return (riLeft.s) < (riRight.s);
  }
};

// map<long long,long long> prime_f(long long n){
//     map<long long,long long>res;
//     for(int i=2;i*i<=n;i++){
//         while(n%i==0){
//             ++res[i];
//             n/=i;
//         }
//     }
//     if(n!=1)res[n]=1;
//     return res;
// }

// int n;
////int dat[2*10000000];
////int dat2[2*10000000];
// int dat[10];
// int dat2[10];
//
// void init(int n_){
//     n=1;
//     while(n<n_)n*=2;
//     for(int i=0;i<2*n-1;i++){
//         dat[i]=0;
//         dat2[i]=0;
//     }
// }
//
// void initset(int k,int a){
//     k+=n-1;
//     dat[k]=a;
//     while(k>0){
//         k=(k-1)/2;
//         dat[k]=dat[k*2+1]+dat[k*2+2];
//     }
// }
//
////[a,b)の間を[l,r]区間で比較しアップデート
////引数のindexに注意
////nは固定。initで計算すみ
////update2(L[i],R[i]+1,0,0,n,D[i]);
// void update2(int a,int b,int k,int l,int r,int v){//v更新値、区間は0-index
//     if(r<=a || b<=l)return;
//     if(a<=l && r<=b){
//         dat[k]+=dat2[k];
//         if(r-l>1){
//             dat2[k*2+1]+=dat2[k]/2;
//             dat2[k*2+1]+=dat2[k]/2;
//         }
//         dat2[k]=v*(r-l);
//         return;
//     }else{
//         update2(a,b,k*2+1,l,(l+r)/2,v);
//         update2(a,b,k*2+2,(l+r)/2,r,v);
//         return;
//     }
// }
//
// int query(int a,int b,int k,int l,int r){
//     if(r<=a || b<=l)return 0;
//     if(a<=l && r<=b){
//         dat[k]+=dat2[k];
//         if(r-l>1){
//             dat2[k*2+1]+=dat2[k]/2;
//             dat2[k*2+1]+=dat2[k]/2;
//         }
//         dat2[k]=0;
//         return dat[k];
//     }
//     else{
//         int vl=query(a,b,k*2+1,l,(l+r)/2);
//         int vr=query(a,b,k*2+2,(l+r)/2,r);
//         return vl+vr;
//     }
// }

// void printb(unsigned int v) {
//     unsigned int mask = (int)1 << (sizeof(v) * CHAR_BIT - 1);
//     do putchar(mask & v ? '1' : '0');
//     while (mask >>= 1);
// }

#ifdef DEBUG

#else

#endif
set<int> G1[100001];
set<int> G2[100001];
ll oya[100001];
ll nak[100001];
ll non[100001];

int ppar[100001];  // 親(1-index)
int rrank[100001]; // 木の深さ

void init(int n) { //
  for (int i = 1; i <= n; i++) {
    ppar[i] = i;
    rrank[i] = 0;
  }
}

int find(int x) {
  if (ppar[x] == x) {
    return x;
  } else {
    return ppar[x] = find(ppar[x]);
  }
}

void unite(int x, int y) {
  x = find(x);
  y = find(y);
  if (x == y) {
    return;
  }
  if (rrank[x] < rrank[y]) {
    ppar[x] = y;
  } else {
    ppar[y] = x;
    if (rrank[x] == rrank[y]) {
      rrank[x]++;
    }
  }
}
bool same(int x, int y) { return find(x) == find(y); }

int main(int argc, const char *argv[]) {
  // scanf("%s",S);
  // scanf("%d",&N);
  // scanf("%lld %lld",&target1,&target2);
  // sscanf(tmp.c_str(),"%dd%d%d",&time[i], &dice[i], &z[i]);
  // getline(cin, target);
  // ifstream ifs("01");//テスト用
  // ifs >> a;
  // ここから

  // 入力高速化
  ios::sync_with_stdio(false);
  cin.tie(0);

  int N, M, K;
  cin >> N >> M >> K;
  init(N);
  for (int i = 0; i < M; i++) {
    int t1, t2;
    cin >> t1 >> t2;
    G1[t1].insert(t2);
    G1[t2].insert(t1);
    unite(t1, t2);
  }
  for (int i = 0; i < K; i++) {
    int t1, t2;
    cin >> t1 >> t2;
    if (same(t1, t2)) {
      G2[t1].insert(t2);
      G2[t2].insert(t1);
    }
  }
  for (int i = 1; i <= N; i++) {
    int t = find(i);
    oya[t]++;
  }

  for (int i = 1; i <= N; i++) {
    int ans = 0;
    int t = find(i);
    ans += oya[t] - 1;
    ans -= G1[i].size();
    ans -= G2[i].size();
    cout << ans << " ";
  }
  cout << endl;

  // ここまで
  // cout << "ans" << endl;
  // cout << " " << "ans" << endl;
  // printf("%.0f\n",ans);//小数点以下表示なし
  // printf("%.7f\n",p);

  return 0;
}