結果

問題 No.1290 Addition and Subtraction Operation
ユーザー kyort0n
提出日時 2020-08-11 05:16:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 3,074 bytes
コンパイル時間 1,971 ms
コンパイル使用メモリ 179,392 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2024-07-22 20:13:33
合計ジャッジ時間 5,664 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 85
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> l_l;
typedef pair<int, int> i_i;
template<class T>
inline bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}

template<class T>
inline bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return true;
    }
    return false;
}

const long long INF = 1e18;
//const ll mod = 1000000007;
ll N, M;
ll B[100500];
ll C[100500];
ll L[105000], R[105000];
struct UnionFind {
    vector<int> par;
    vector<int> rank;
    vector<ll> Size;
    UnionFind(int n = 1) {
        init(n);
    }

    void init(int n = 1) {
        par.resize(n + 1); rank.resize(n + 1); Size.resize(n + 1);
        for (int i = 0; i <= n; ++i) par[i] = i, rank[i] = 0, Size[i] = 1;
    }

    int root(int x) {
        if (par[x] == x) {
            return x;
        }
        else {
            int r = root(par[x]);
            return par[x] = r;
        }
    }

    bool issame(int x, int y) {
        return root(x) == root(y);
    }

    bool merge(int x, int y) {
        x = root(x); y = root(y);
        if (x == y) return false;
        if (rank[x] < rank[y]) swap(x, y);
        if (rank[x] == rank[y]) ++rank[x];
        par[y] = x;
        Size[x] += Size[y];
        return true;
    }

    ll size(int x){
        return Size[root(x)];
    }
};

ll sum[105000];
ll subsum[105000];

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    cin >> N >> M;
    for(int i = 1; i <= N; i++) cin >> B[i];
    for(int i = 0; i <= N; i++) {
        C[i] = B[i] + B[i+1];
    }
    N++;
    UnionFind vertex(N);
    for(int i = 0; i < M; i++) {
        cin >> L[i] >> R[i];
        L[i]--;
        if((R[i] - L[i]) % 2 == 0) {
            vertex.merge(L[i], R[i]);
        }
    }
    UnionFind bipartite(2 * N), connected(N);
    for(int i = 0; i < M; i++) {
        if((R[i] - L[i]) % 2 == 1) {
            ll a = vertex.root(L[i]);
            ll b = vertex.root(R[i]);
            bipartite.merge(a + N, b);
            bipartite.merge(a, b + N);
            connected.merge(a, b);
        }
    }
    for(int i = 0; i < N; i++) {
        sum[connected.root(vertex.root(i))] += C[i];
        if(bipartite.issame(vertex.root(i), connected.root(vertex.root(i)))) subsum[connected.root(vertex.root(i))] += C[i];
    }
    /*
    for(int i = 0; i < N; i++) {
        if(vertex.root(i) == i) cerr << i << " ";
    }
    cerr << endl;
    */
    string ans = "YES";
    for(int i = 0; i < N; i++) {
        if(vertex.root(i) != i) continue;
        if(connected.root(i) != i) continue;
        //cerr << i << " " << sum[i] << " " << subsum[i] << " " << bipartite.issame(i, i + N) << endl;
        if(bipartite.issame(i, i + N)) {
            if(abs(sum[i]) % 2 != 0) {
                //cerr << i << endl;
                ans = "NO";
            }
        } else {
            if(subsum[i] * 2 != sum[i]) {
                //cerr << i << endl;
                ans = "NO";
            }
        }
    }
    cout << ans << endl;
    return 0;
}
0