結果

問題 No.1266 7 Colors
ユーザー keikei
提出日時 2020-10-23 22:43:01
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 236 ms / 3,000 ms
コード長 4,013 bytes
コンパイル時間 2,013 ms
コンパイル使用メモリ 176,980 KB
実行使用メモリ 14,964 KB
最終ジャッジ日時 2024-07-21 11:35:21
合計ジャッジ時間 7,250 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 170 ms
5,888 KB
testcase_04 AC 212 ms
11,264 KB
testcase_05 AC 177 ms
6,400 KB
testcase_06 AC 225 ms
12,160 KB
testcase_07 AC 230 ms
13,312 KB
testcase_08 AC 227 ms
11,392 KB
testcase_09 AC 209 ms
9,728 KB
testcase_10 AC 204 ms
9,216 KB
testcase_11 AC 178 ms
6,940 KB
testcase_12 AC 187 ms
7,680 KB
testcase_13 AC 192 ms
8,320 KB
testcase_14 AC 172 ms
6,400 KB
testcase_15 AC 236 ms
13,440 KB
testcase_16 AC 185 ms
7,808 KB
testcase_17 AC 235 ms
13,056 KB
testcase_18 AC 196 ms
14,964 KB
testcase_19 AC 72 ms
14,520 KB
testcase_20 AC 72 ms
14,548 KB
testcase_21 AC 165 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int INF = 1e9;
const ll LINF = 1e18;
inline ll gcd(ll a, ll b) { return b ? gcd(b, a%b) : a; }
inline ll lcm(ll a, ll b) { return a / gcd(a, b)*b; }
template<class S,class T> ostream& operator << (ostream& out,const pair<S,T>& o){ out << "(" << o.first << "," << o.second << ")"; return out; }
template<class T> ostream& operator << (ostream& out,const vector<T>& V){ for(int i = 0; i < V.size(); i++){ out << V[i]; if(i!=V.size()-1) out << " ";} return out; }
template<class T> ostream& operator << (ostream& out,const vector<vector<T> >& Mat){ for(int i = 0; i < Mat.size(); i++) { if(i != 0) out << endl; out << Mat[i];} return out; }
template<class S,class T> ostream& operator << (ostream& out,const map<S,T>& mp){ out << "{ "; for(auto it = mp.begin(); it != mp.end(); it++){ out << it->first << ":" << it->second; if(mp.size()-1 != distance(mp.begin(),it)) out << ", "; } out << " }"; return out; }
template<typename T>vector<T> make_v(size_t a){return vector<T>(a);}
template<typename T,typename... Ts>auto make_v(size_t a,Ts... ts){return vector<decltype(make_v<T>(ts...))>(a,make_v<T>(ts...));}
template<typename T,typename V> typename enable_if<is_class<T>::value==0>::type fill_v(T &t,const V &v){t=v;}
template<typename T,typename V> typename enable_if<is_class<T>::value!=0>::type fill_v(T &t,const V &v){for(auto &e:t) fill_v(e,v);}
/*
 <url:>
 問題文============================================================
 =================================================================
 解説=============================================================
 ================================================================
 */

/*
 UnionFind
 */
struct UnionFind {
    vector<int> data;
    UnionFind(int size) : data(size, -1) { }
    bool unionSet(int x, int y) {
        x = root(x); y = root(y);
        if (x != y) {
            if (data[y] < data[x]) swap(x, y);
            data[x] += data[y]; data[y] = x;
        }
        return x != y;
    }
    bool findSet(int x, int y) {
        return root(x) == root(y);
    }
    int root(int x) {
        return data[x] < 0 ? x : data[x] = root(data[x]);
    }
    int size(int x) {
        return -data[root(x)];
    }
};

template<class Type>
Type solve(Type res = Type()){
    int N,M,Q; cin >> N >> M >> Q;

    auto convert_id = [&](int v,int c){
        return v*7 + c;
    };

    UnionFind UF(N*7);

    vector<string> state(N);

    for(int i = 0; i < N;i++){
        cin >> state[i];

        for(int j = 0; j < 7;j++){
            int k = (j+1)%7;

            if(state[i][j] == '1' && state[i][k] == '1'){
                UF.unionSet(convert_id(i,j),convert_id(i,k));
            }
        }
    }

    vector<vector<int>> G(N);
    for(int i = 0; i < M;i++){
        int u,v; cin >> u >> v;
        u--; v--;
        G[u].emplace_back(v);
        G[v].emplace_back(u);

        for(int j = 0; j < 7;j++){
            if(state[u][j] == '1' && state[v][j] == '1'){
                UF.unionSet(convert_id(u,j),convert_id(v,j));
            }
        }
    }

    while(Q--){
        int com,x,y;
        cin >> com >> x >> y;
        x--; y--;

        if(com == 1){
            state[x][y] = '1';

            int l = (y+6)%7;
            if(state[x][l] == '1'){
                UF.unionSet(convert_id(x,y),convert_id(x,l));
            }
            int r = (y+1)%7;
            if(state[x][r] == '1'){
                UF.unionSet(convert_id(x,y),convert_id(x,r));
            }

            for(int v:G[x]){
                if(state[v][y] == '1'){
                    UF.unionSet(convert_id(x,y),convert_id(v,y));
                }
            }

        }else{

            cout << UF.size(convert_id(x,0)) << endl;
        }
    }
    return res;
}
int main(void) {
    cin.tie(0); ios::sync_with_stdio(false);
    solve<ll>(0);
    // cout << fixed << setprecision(12) << solve<ll>() << endl;
    return 0;
}
0