#959984 (C++17) No.2674 k-Walk on Bipartite

提出ソース

結果

問題	No.2674 k-Walk on Bipartite
ユーザー	ococonomy1
提出日時	2024-03-15 21:38:20
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	132 ms / 2,000 ms
コード長	5,747 bytes
コンパイル時間	1,388 ms
コンパイル使用メモリ	124,784 KB
実行使用メモリ	15,080 KB
最終ジャッジ日時	2024-03-15 21:38:30
合計ジャッジ時間	4,535 ms
ジャッジサーバーID （参考情報）	judge15 / judge14

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,676 KB
testcase_01	AC	2 ms 6,676 KB
testcase_02	AC	2 ms 6,676 KB
testcase_03	AC	2 ms 6,676 KB
testcase_04	AC	2 ms 6,676 KB
testcase_05	AC	2 ms 6,676 KB
testcase_06	AC	2 ms 6,676 KB
testcase_07	AC	78 ms 14,100 KB
testcase_08	AC	94 ms 11,500 KB
testcase_09	AC	67 ms 13,368 KB
testcase_10	AC	121 ms 13,188 KB
testcase_11	AC	67 ms 10,996 KB
testcase_12	AC	113 ms 11,624 KB
testcase_13	AC	62 ms 12,060 KB
testcase_14	AC	20 ms 9,928 KB
testcase_15	AC	132 ms 14,220 KB
testcase_16	AC	91 ms 14,736 KB
testcase_17	AC	88 ms 11,256 KB
testcase_18	AC	34 ms 9,460 KB
testcase_19	AC	76 ms 12,628 KB
testcase_20	AC	68 ms 13,764 KB
testcase_21	AC	102 ms 12,140 KB
testcase_22	AC	128 ms 15,080 KB
testcase_23	AC	2 ms 6,676 KB
testcase_24	AC	2 ms 6,676 KB
testcase_25	AC	2 ms 6,676 KB
testcase_26	AC	2 ms 6,676 KB
testcase_27	AC	2 ms 6,676 KB
testcase_28	AC	1 ms 6,676 KB
testcase_29	AC	2 ms 6,676 KB
testcase_30	AC	1 ms 6,676 KB
testcase_31	AC	2 ms 6,676 KB
testcase_32	AC	2 ms 6,676 KB
testcase_33	AC	2 ms 6,676 KB
testcase_34	AC	2 ms 6,676 KB
testcase_35	AC	2 ms 6,676 KB
testcase_36	AC	2 ms 6,676 KB
testcase_37	AC	2 ms 6,676 KB
testcase_38	AC	2 ms 6,676 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <climits>
#include <deque>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <tuple>
#include <vector>
using namespace std;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using pli = pair<ll,int>;
#define TEST cerr << "TEST" << endl
#define AMARI 998244353
//#define AMARI 1000000007
#define el '\n'
#define El '\n'

class ococo_unionfind {
    // できること
    // 点の挿入
    // その点の根を求める関数
    // 辺の挿入
    // 連結判定
    // 島が何個あるかの出力
    // それぞれの島について、何個の点があるかの出力
  public:
    ococo_unionfind(int n = 0) {
        for(int i = 0; i < n; i++) vinsert();
    }
    int simakosuu = 0;
    // g[i] = {その点の一個上の点,その点のrank}
    // その点のrank:その点の下に何個点があるか（上に何個あるかに変えた方がいいかも？）
    vector<pair<int, int>> g;

    // rs[i] = その点が含まれている連結成分に何個の点があるか
    // その連結成分の根について聞かないと返さない
    vector<int> rs;
    // 点の挿入 O(1)
    void vinsert(void) {
        g.emplace_back(g.size(), 1);
        simakosuu++;
        rs.push_back(1);
    }
    // ある点の根を求める関数 O(α(N))
    int ne(int a) {
        if(g[a].first == a) return a;
        else {
            return g[a].first = ne(g[a].first);
        }
    }
    // 辺の挿入 O(logN)
    void einsert(int a, int b) {
        if(a != b) {
            int a1 = ne(a), a2 = ne(b);
            if(a1 != a2) {
                simakosuu--;
                int rs12sum = rs[a1] + rs[a2];
                rs[a1] = rs12sum;
                rs[a2] = rs12sum;
                if(g[a1].second < g[a2].second) {
                    g[a1].first = a2;
                    g[a2].second = max(g[a1].second + 1, g[a2].second);
                } else {
                    g[a2].first = a1;
                    g[a1].second = max(g[a2].second + 1, g[a1].second);
                }
            }
        }
    }

    void peinsert(pair<int,int> p){
        einsert(p.first,p.second);
        return;
    }

    // 2つのノードが繋がっているか判定する関数 O(α(N))
    bool renketucheck(int a, int b) {
        if(ne(a) == ne(b)) return true;
        else return false;
    }
    bool prenketucheck(pair<int,int> p){
        return renketucheck(p.first,p.second);
    }
    // 何個の島に分かれているか出力する関数 O(1)
    int islandnum(void) {
        return simakosuu;
    }
    // ある点について、その点が含まれている連結成分が何個の点を持つか返す関数
    // O(α(N))
    int islandsize(int a) {
        return rs[ne(a)];
    }
};


#define MULTI_TEST_CASE false
void solve(void){
    //問題を見たらまず「この問題設定から言えること」をいっぱい言う
    //一個回答に繋がりそうな解法が見えても、実装や細かい詰めに時間がかかりそうなら別の方針を考えてみる
    //添え字回りで面倒になりそうなときは楽になる言い換えを実装の前にじっくり考える
    //ある程度考察しても全然取っ掛かりが見えないときは実験をしてみる
    //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く

    //F(Eから辺をいくつか削除したもの)が与えられる
    //Eについてs,tをつなぐ長さ k の歩道（ウォーク？）があるか判定せよ
    //ただし、E,Fは共に二部グラフ

    int n,m;
    int s,t,k;
    cin >> n >> m >> s >> t >> k;
    s--; t--;
    vector<vector<int>> g(n);
    ococo_unionfind uf(n);
    while(m--){
        int u,v;
        cin >> u >> v;
        u--; v--;
        g[u].push_back(v);
        g[v].push_back(u);
        uf.einsert(u,v);
    }
    if(n == 1){
        cout << "No" << El;
        return;
    }
    if(n == 2){
        //始点と終点が一緒で既にもう一方との点とつながってる場合、どうやっても Yes
        if(((s == t) == (k % 2 == 0)) && uf.islandnum() == 1){
            cout << "Yes" << el;
            return;
        }
        if((s == t) == (k % 2 == 0)){
            cout << "Unknown" << el;
            return;
        }
        cout << "No" << el;
        return;
    }
    if(s == t){
        if(k % 2 == 0 && uf.islandsize(s) != 1)cout << "Yes" << el;
        if(k % 2 == 0 && uf.islandsize(s) == 1)cout << "Unknown" << el;
        if(k % 2 == 1) cout << "No" << el;
        return;
    }
    if(!uf.renketucheck(s,t)){
        cout << "Unknown" << el;
        return;
    }
    vector<int> dist(n,-1);
    dist[s] = 0;
    queue<int> que; que.push(s);
    while(!que.empty()){
        int fr = que.front();
        que.pop();
        for(int i = 0; i < g[fr].size(); i++){
            int to = g[fr][i];
            if(dist[to] != -1)continue;
            dist[to] = dist[fr] + 1;
            que.push(to);
        }
    }

    if(dist[t] % 2 != k % 2){
        cout << "No" << el;
        return;
    }
    if(dist[t] <= k){
        cout << "Yes" << el;
        return;
    }
    cout << "Unknown" << el;    
    return;
}

void calc(void){
    return;
}

signed main(void){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    calc();
    int t = 1;
    if(MULTI_TEST_CASE)cin >> t;
    while(t--){
        solve();
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード