結果

問題 No.2674 k-Walk on Bipartite
コンテスト
ユーザー ococonomy1
提出日時 2024-03-15 21:38:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 129 ms / 2,000 ms
コード長 5,747 bytes
コンパイル時間 1,348 ms
コンパイル使用メモリ 120,392 KB
最終ジャッジ日時 2025-02-20 04:45:06
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#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;
}
0