結果

問題 No.2291 Union Find Estimate
ユーザー srjywrdnprktsrjywrdnprkt
提出日時 2023-05-13 02:31:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,244 bytes
コンパイル時間 1,148 ms
コンパイル使用メモリ 115,368 KB
実行使用メモリ 7,764 KB
最終ジャッジ日時 2024-05-06 15:07:31
合計ジャッジ時間 3,318 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 315 ms
5,376 KB
testcase_03 AC 10 ms
7,764 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 20 ms
5,376 KB
testcase_11 AC 35 ms
5,376 KB
testcase_12 AC 66 ms
5,376 KB
testcase_13 AC 6 ms
5,376 KB
testcase_14 AC 10 ms
5,376 KB
testcase_15 AC 8 ms
5,576 KB
testcase_16 AC 8 ms
5,376 KB
testcase_17 AC 6 ms
5,376 KB
testcase_18 AC 5 ms
5,376 KB
testcase_19 AC 21 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cmath>
#include <map>
#include <set>
#include <iomanip>
#include <queue>
#include <algorithm>
#include <numeric>
#include <deque>

using namespace std;
 
struct UnionFind {
    long long ngroup;
    vector<long long> par;
    vector<long long> siz;

    UnionFind(long long N) : par(N), siz(N) {
        ngroup = N;
        for(long long i = 0; i < N; i++){
            par[i] = i;
            siz[i] = 1;
        }   
    }   
 
    long long root(long long x) {
        if (par[x] == x) return x;
        return par[x] = root(par[x]);
    }   
 
    long long unite(long long x, long long y) {
        long long rx = root(x);
        long long ry = root(y);
        if (rx == ry) return rx; 
        ngroup--;
        if (siz[rx] > siz[ry]) swap(rx, ry);
        par[rx] = ry; 
        siz[ry] += siz[rx];
        return ry; 
    }   
 
    bool same(long long x, long long y) {
        long long rx = root(x);
        long long ry = root(y);
        return rx == ry; 
    }   
 
    long long size(long long x){ 
        return siz[root(x)];
    }   

    long long group_count(){
        return ngroup;
    }
};

const long long m = 998244353;
long long mod_exp(long long b, long long e){
    if (e > 0 && b == 0) return 0;
    long long ans = 1;
    b %= m;

    while (e > 0){
        if ((e & 1LL)) ans = (ans * b) % m;
        e = e >> 1LL;
        b = (b*b) % m;
    }

    return ans;
}


int main(){
 
    long long W, H;
    string S;
    cin >> W >> H;
    UnionFind tree(W+10);

    for (int i=0; i<H; i++){
        cin >> S;
        bool f=1;
        vector<vector<int>> v(26);
        for (int j=0; j<W; j++){
            if ('a' <= S[j] && S[j] <= 'z') v[S[j]-'a'].push_back(j);
            else if ('0' <= S[j] && S[j] <= '9') tree.unite(j, S[j]-'0'+W);
        }
        for (int j=0; j<26; j++){
            if (v[j].size() == 0) continue;
            for (auto x : v[j]) tree.unite(v[j][0], x);
        }

        for (int j=0; j<10; j++){
            for (int k=j+1; k<10; k++) if (tree.same(W+j, W+k)){
                cout << 0 << endl;
                f=0;
            }
        }
        if (f) cout << mod_exp(10, tree.group_count()-10) << endl;
    }

    return 0;
}
0