提出 #18067389


ソースコード 拡げる

Copy
//#define _GLIBCXX_DEBUG
//#include "atcoder/all"
//using namespace atcoder;
#include <bits/stdc++.h>
#define int long long
#define ll long long
using ull = unsigned long long;
using namespace std;
#define Dump(x)                             \
    if (dbg) {                              \
        cerr << #x << " = " << (x) << endl; \
    }
#define overload4(_1, _2, _3, _4, name, ...) name
#define FOR1(n) for (ll i = 0; i < (n); ++i)
#define FOR2(i, n) for (ll i = 0; i < (n); ++i)
#define FOR3(i, a, b) for (ll i = (a); i < (b); ++i)
#define FOR4(i, a, b, c) for (ll i = (a); i < (b); i += (c))
#define FOR(...) overload4(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__)
#define FORR(i, a, b) for (int i = (a); i <= (b); ++i)
#define bit(n, k) (((n) >> (k)) & 1) /*nのk bit目*/
namespace mydef {
const int INF = 1ll << 60;
const int MOD = 1e9 + 7;
template <class T>
bool chmin(T& a, const T& b) {
    if (a > b) {
        a = b;
        return 1;
    } else
        return 0;
}
template <class T>
bool chmax(T& a, const T& b) {
    if (a < b) {
        a = b;
        return 1;
    } else
        return 0;
}
void Yes(bool flag = true) {
    if (flag)
        cout << "Yes" << endl;
    else
        cout << "No" << endl;
}
void No(bool flag = true) {
    Yes(!flag);
}
void YES(bool flag = true) {
    if (flag)
        cout << "YES" << endl;
    else
        cout << "NO" << endl;
}
void NO(bool flag = true) {
    YES(!flag);
}
template <typename A, size_t N, typename T>
void Fill(A (&array)[N], const T& val) {
    std::fill((T*)array, (T*)(array + N), val);
}
bool dbg = true;
}  // namespace mydef
using namespace mydef;
#define pb push_back
//#define mp make_pair
#define eb emplace_back
#define lb lower_bound
#define ub upper_bound
#define all(v) (v).begin(), (v).end()
#define SZ(x) ((int)(x).size())
#define vi vector<int>
#define vvi vector<vector<int>>
#define vp vector<pair<int, int>>
#define vvp vector<vector<pair<int, int>>>
#define pi pair<int, int>
//#define P pair<int, int>
//#define V vector<int>
//#define S set<int>
#define asn ans

//UnionFind
//How to use
//unite(x,y) : 集合 x と y を併合する. 併合済のとき false, 未併合のとき true が返される
//find(k) : 要素 k が属する集合を求める
//size(k) : 要素 k が属する集合の要素の数を求める
struct UnionFind {
    vector<int> data;

    UnionFind(int sz) {
        data.assign(sz, -1);
    }

    bool unite(int x, int y) {
        x = find(x), y = find(y);
        if (x == y)
            return (false);
        if (data[x] > data[y])
            swap(x, y);
        data[x] += data[y];
        data[y] = x;
        return (true);
    }

    int find(int k) {
        if (data[k] < 0)
            return (k);
        return (data[k] = find(data[k]));
    }

    bool same(int x, int y) {
        return find(x) == find(y);
    }

    int size(int k) {
        return (-data[find(k)]);
    }
};

int N, M, Q;
vp con;
vp discon;

void solve() {
    UnionFind uf(N);
    for (auto& [a, b] : con) {
        uf.unite(a, b);
    }
    set<int> roots;
    for (int i = 0; i < N; i++) {
        roots.insert(uf.find(i));
    }
    bool flag = true;
    for (auto& [a, b] : discon) {
        if (uf.same(a, b))
            flag = false;
    }
    int cnt = roots.size();
    if (!flag)
        No();
    else if (M == N - 1 && cnt == 1)
        Yes();
    else if (M == N - 1 && discon.size() == 0)
        Yes();
    else if (M == N - 1)
        No();
    else {
        int MAX = (N - cnt) + cnt * (cnt - 1) / 2;
        if (M > MAX)
            No();
        else if (cnt == 2)
            No();
        else
            Yes();
    }
}

signed main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    cin >> N >> M >> Q;
    while (Q--) {
        int a, b, c;
        cin >> a >> b >> c;
        if (c == 0) {
            con.push_back({a, b});
        } else
            discon.push_back({a, b});
    }

    solve();
    return 0;
}

提出情報

提出日時
問題 D - Unique Path
ユーザ cuthbert
言語 C++ (GCC 9.2.1)
得点 700
コード長 3975 Byte
結果 AC
実行時間 53 ms
メモリ 10332 KB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 700 / 700
結果
AC × 3
AC × 39
セット名 テストケース
Sample sample-01.txt, sample-02.txt, sample-03.txt
All 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt, 01-15.txt, 01-16.txt, 01-17.txt, 01-18.txt, 01-19.txt, 01-20.txt, 01-21.txt, 01-22.txt, 01-23.txt, 01-24.txt, 01-25.txt, 01-26.txt, 01-27.txt, 01-28.txt, 01-29.txt, 01-30.txt, 01-31.txt, 01-32.txt, 01-33.txt, 01-34.txt, 01-35.txt, 01-36.txt, sample-01.txt, sample-02.txt, sample-03.txt
ケース名 結果 実行時間 メモリ
01-01.txt AC 11 ms 3572 KB
01-02.txt AC 3 ms 3612 KB
01-03.txt AC 39 ms 5332 KB
01-04.txt AC 34 ms 5256 KB
01-05.txt AC 35 ms 5160 KB
01-06.txt AC 36 ms 6792 KB
01-07.txt AC 43 ms 7824 KB
01-08.txt AC 47 ms 8828 KB
01-09.txt AC 41 ms 7876 KB
01-10.txt AC 37 ms 6068 KB
01-11.txt AC 38 ms 5996 KB
01-12.txt AC 40 ms 6012 KB
01-13.txt AC 37 ms 6072 KB
01-14.txt AC 32 ms 5520 KB
01-15.txt AC 36 ms 5544 KB
01-16.txt AC 28 ms 5520 KB
01-17.txt AC 36 ms 5384 KB
01-18.txt AC 32 ms 5292 KB
01-19.txt AC 34 ms 5280 KB
01-20.txt AC 35 ms 5492 KB
01-21.txt AC 30 ms 5504 KB
01-22.txt AC 31 ms 5544 KB
01-23.txt AC 50 ms 10228 KB
01-24.txt AC 51 ms 10332 KB
01-25.txt AC 52 ms 10200 KB
01-26.txt AC 52 ms 10172 KB
01-27.txt AC 53 ms 7960 KB
01-28.txt AC 53 ms 7852 KB
01-29.txt AC 52 ms 8048 KB
01-30.txt AC 50 ms 7912 KB
01-31.txt AC 52 ms 7888 KB
01-32.txt AC 49 ms 7860 KB
01-33.txt AC 41 ms 6368 KB
01-34.txt AC 50 ms 7776 KB
01-35.txt AC 33 ms 5516 KB
01-36.txt AC 33 ms 5516 KB
sample-01.txt AC 2 ms 3612 KB
sample-02.txt AC 2 ms 3496 KB
sample-03.txt AC 2 ms 3620 KB