結果

問題 No.2291 Union Find Estimate
ユーザー srjywrdnprkt
提出日時 2023-05-13 02:51:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 286 ms / 2,000 ms
コード長 2,179 bytes
コンパイル時間 1,134 ms
コンパイル使用メモリ 113,448 KB
最終ジャッジ日時 2025-02-12 23:48:13
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

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);

    while(H){
        H--;
        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)) f=0;
        }
        cout << (f ? mod_exp(10, tree.group_count()-10) : 0) << endl;
    }

    return 0;
}
0