結果

問題 No.482 あなたの名は
ユーザー wakapaijazzwakapaijazz
提出日時 2020-10-01 01:16:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 65 ms / 2,000 ms
コード長 1,527 bytes
コンパイル時間 2,024 ms
コンパイル使用メモリ 199,864 KB
最終ジャッジ日時 2025-01-14 23:43:01
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

#define rep(i,n) for(int (i)=0;(i)<(n);(i)++)
#define rep3(i,m,n) for(int (i)=m;(i)<=(n);(i)++)
#define rep3rev(i,m,n) for(int (i)=m;(i)>=(n);(i)--)
#define all(a) (a.begin()),(a.end())
#define rall(a) (a.rbegin()),(a.rend())
#define fi first
#define se second
#define pb push_back
#define eb emplace_back

using ll = long long;
using vll = vector<ll>;
using vi = vector<int>;
using vvi = vector<vector<int>>;
using P = pair<int, int>;

struct UnionFind {
    vector<int> d;
    UnionFind(int N) : d(N, -1){}

    int root(int x) {
        if (d[x] < 0) return x;
        return d[x] = root(d[x]);
    }

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

    bool same(int x, int y) {
        return root(x) == root(y);
    }
    
    int size(int x) {
        return -d[root(x)];
    }
};

void Main(){
    ll n, k; cin >> n >> k;
    UnionFind UF(n);
    rep(i, n){
        int d; cin >> d; d--;
        if(i != d){
            UF.unite(i, d);
        }
    }
    set<int> Root;
    rep(i, n){
        int r = UF.root(i);
        Root.insert(r);
    }
    ll ans = n - Root.size();
    if(k < ans || (k - ans) % 2) cout << "NO" << endl;
    else cout << "YES" << endl; 
    return;
}

int main(){
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);
    cout << fixed << setprecision(15);
    Main();
    return 0;
}
0