結果

問題 No.1266 7 Colors
ユーザー recososo
提出日時 2020-10-23 22:53:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,851 bytes
コンパイル時間 1,908 ms
コンパイル使用メモリ 175,248 KB
実行使用メモリ 17,708 KB
最終ジャッジ日時 2024-07-21 12:03:25
合計ジャッジ時間 6,872 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
#define REP(i, n) for (int i = 0; i < (n); ++i)
#define REPR(i, n) for (int i = n - 1; i >= 0; --i)
#define FOR(i, m, n) for (int i = m; i < n; ++i)
#define FORR(i, m, n) for (int i = m; i >= n; --i)
#define ALL(v) (v).begin(),(v).end()
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
const ll INF=1LL<<60;
const int inf=(1<<30)-1;
const int mod=1e9+7;
int dx[8]={1,0,-1,0,-1,-1,1,1};
int dy[8]={0,1,0,-1,-1,1,-1,1};
struct UnionFind {
    vector<int> par,siz; // par[i]:iの親の番号　(例) par[3] = 2 : 3の親が2
    UnionFind(int N) : par(N) , siz(N) { //最初は全てが根であるとして初期化
        for(int i = 0; i < N; i++) par[i] = i;
        for(int i = 0; i < N; i++) siz[i] = 1;
    }
    int root(int x) { // データxが属する木の根を再帰で得る：root(x) = {xの木の根}
        if (par[x] == x) return x;
        return par[x] = root(par[x]);
    }
    void unite(int x, int y) { // xとyの木を併合
        int rx = root(x); //xの根をrx
        int ry = root(y); //yの根をry
        if (rx == ry) return; //xとyの根が同じ(=同じ木にある)時はそのまま
        par[ry] = rx; //xとyの根が同じでない(=同じ木にない)時：xの根rxをyの根ryにつける
        siz[rx]+=siz[ry];
    }
    bool same(int x, int y) { // 2つのデータx, yが属する木が同じならtrueを返す
        int rx = root(x);
        int ry = root(y);
        return rx == ry;
    }
    int size(int x){
        return siz[root(x)];
    }
};
int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    int n,m,q;cin >> n >> m >> q;
    vector<string> s(n);
    UnionFind uf(n*7);
    REP(i,n){
        cin >> s[i];
    }
    REP(i,n){
        REP(j,7){
            if(s[i][j]=='1'&&s[i][(j+1)%7]=='1'){
                uf.unite(i*7+j,i*7+(j+1)%7);
            }
        }
    }
    vector<vector<int>> e(n);
    REP(i,m){
        int u,v;cin >> u >> v;
        u--,v--;
        e[u].push_back(v);
        e[v].push_back(u);
        REP(j,7){
            if(s[u][j]=='1'&&s[v][j]=='1'){
                uf.unite(u*7+j,v*7+j);
            }
        }
    }
    REP(i,q){
        int t,x,y;cin >> t >> x >> y;
        x--,y--;
        if(t==1){
            if(s[x][(y+6)%7]=='1'){
                uf.unite(x*7+(y+6)%7,x*7+y);
            }
            if(s[x][(y+1)%7]=='1'){
                uf.unite(x*7+(y+1)%7,x*7+y);
            }
            for(int to:e[x]){
                if(s[to][y]=='1'){
                    uf.unite(to*7+y,x*7+y);
                }
            }
        }
        else{
            cout << uf.size(x*7) << endl;
        }
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0