AtCoder Typical Contest 001

Submission #7128463

Source codeソースコード

#include <iostream>
#include <cstdlib>
#include <vector>
#include <stack>
#include <string>
#include <climits>
#include <algorithm>
#include <functional>
#include <string.h>
#include <numeric>
#include <math.h>

#define LOOP(N) for(int i=0; i<(N); ++i)
#define REP(i, N) for(int i=0; i<(N); ++i)
#define FOR(i, start, end) for(int i=(start); i<(end); ++i)
#define ALL(a) (a).begin(),(a).end()

using namespace std;

using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
using Graph = vector< vector<int> >;

void sayYes() {cout << "Yes" << endl;}
void sayNo() {cout << "No" << endl;}


class UnionFind {
    vector<int> parents;
    
public:
    UnionFind(int N) : parents(N) {
        REP(i, N) parents[i] = i;
    }
    int getRoot(int v) {
        return parents[v] == v ? v : parents[v] = getRoot(parents[v]);
    }
    void unite(int a, int b) {
        a = getRoot(parents[a]);
        b = getRoot(parents[b]);
        parents[a] = b;
    }
    bool isSameGroup(int a, int b) {
        return getRoot(a) == getRoot(b);
    }
};


int getRoot(vector<int> &lines, int v) {
    return lines[v] == v ? v : lines[v] = getRoot(lines, lines[v]);
}


int main() {
    int N, Q; cin >> N >> Q;
    UnionFind unf(N);

    int p, a, b;
    REP(i, Q) {
        cin >> p >> a >> b;

        // conbine
        if (p == 0) {
            unf.unite(a, b);
        }
        // judge
        if (p == 1) {
            if (unf.isSameGroup(a, b)) sayYes();
            else sayNo();
            // cout << lines[a] << " " << lines[b] << endl;
        }
    }
}

Submission

Task問題 B - Union Find
User nameユーザ名 ScientistB
Created time投稿日時
Language言語 C++14 (GCC 5.4.1)
Status状態 AC
Score得点 100
Source lengthソースコード長 1649 Byte
File nameファイル名
Exec time実行時間 467 ms
Memory usageメモリ使用量 1280 KB

Test case

Set

Set name Score得点 / Max score Cases
Sample - 00_sample_01.txt
All 100 / 100 00_sample_01.txt,subtask_01_01.txt,subtask_01_02.txt,subtask_01_03.txt,subtask_01_04.txt,subtask_01_05.txt,subtask_01_06.txt,subtask_01_07.txt,subtask_01_08.txt,subtask_01_09.txt,subtask_01_10.txt,subtask_01_11.txt,subtask_01_12.txt,subtask_01_13.txt,subtask_01_14.txt,subtask_01_15.txt,subtask_01_16.txt,subtask_01_17.txt,subtask_01_18.txt

Test case

Case name Status状態 Exec time実行時間 Memory usageメモリ使用量
00_sample_01.txt AC 1 ms 256 KB
subtask_01_01.txt AC 273 ms 640 KB
subtask_01_02.txt AC 1 ms 640 KB
subtask_01_03.txt AC 407 ms 1024 KB
subtask_01_04.txt AC 467 ms 1280 KB
subtask_01_05.txt AC 25 ms 256 KB
subtask_01_06.txt AC 25 ms 640 KB
subtask_01_07.txt AC 427 ms 768 KB
subtask_01_08.txt AC 464 ms 1280 KB
subtask_01_09.txt AC 1 ms 256 KB
subtask_01_10.txt AC 1 ms 640 KB
subtask_01_11.txt AC 422 ms 896 KB
subtask_01_12.txt AC 458 ms 1280 KB
subtask_01_13.txt AC 348 ms 768 KB
subtask_01_14.txt AC 2 ms 640 KB
subtask_01_15.txt AC 417 ms 896 KB
subtask_01_16.txt AC 462 ms 1280 KB
subtask_01_17.txt AC 310 ms 1024 KB
subtask_01_18.txt AC 305 ms 1024 KB