結果

問題 No.1909 Detect from Substrings
ユーザー wudi ma
提出日時 2022-04-24 01:53:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 945 ms / 2,000 ms
コード長 4,071 bytes
コンパイル時間 2,904 ms
コンパイル使用メモリ 199,056 KB
実行使用メモリ 39,792 KB
最終ジャッジ日時 2024-06-25 11:05:21
合計ジャッジ時間 24,180 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:91:23: warning: 'pos1' may be used uninitialized [-Wmaybe-uninitialized]
   91 |     for(int i = 0 ; i < pos1 ; i ++ ){
      |                     ~~^~~~~~
main.cpp:77:9: note: 'pos1' was declared here
   77 |     int pos1,pos2;
      |         ^~~~

ソースコード

diff # 

#pragma GCC optimize("O3")
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
unsigned long long has[N],base[N];

unsigned long long get(int l,int r){
    if(r < l)return 0;
    return has[r] - has[l-1] * base[r-l+1];
}

unordered_map<unsigned long long,int>mp;
int main(){
    base[0] = 1;
    for(int i = 1 ; i < N ; i ++ )base[i] = base[i-1] * 131;
    int n,m;cin >> n >> m;
    vector<string>s(n);
    for(int i = 0 ; i < n ; i ++ ){
        cin >> s[i];
    }
    sort(s.begin(),s.end());
    s.resize(unique(s.begin(),s.end())-s.begin());
    if(s.size() == 1){
        return cout << (m+1) * 26,0;
    }   
    string ta = s[0] + s[1][s[1].size()-1];
    for(int i = 1 ; i <= m+1 ; i ++ ){
        has[i] = has[i-1] * 131 + ta[i-1];
    }
    for(int i = 1 ; i <= m+1 ; i ++ ){
        mp[get(1,i-1) * 131 + get(i+1,m+1)];
    }
    bool ok = 0;
    // cout << get(2,m+1) << '\n';
    mp[get(1,m)] = mp[get(2,m+1)] = 1;
    for(int i = 0 ; i < n ; i ++ ){
        ok = 0;
        for(int j = 1 ; j <= m ; j ++ ){
            has[j] = has[j-1] * 131 + s[i][j-1];
        }
        // cout << get(1,2) << '\n';
        for(int j = 1 ; j <= m ; j ++ ){
            // cout << get(j,m) << '\n';
            if(mp.count(get(1,j-1) * 131 + get(j,m))){
                ok = 1;
                break;
            }
        }
        if(ok == 0)break;
    }
    if(ok){
        string tb = s[1][0] + s[0];
        mp.clear();
        for(int i = 1 ; i <= m+1 ; i ++ ){
            has[i] = has[i-1] * 131 + tb[i-1];
        }
        for(int i = 1 ; i <= m+1 ; i ++ ){
            mp[get(1,i-1) * 131 + get(i+1,m+1)];
        }
        mp[get(1,m)] = mp[get(2,m+1)] = 1;
        for(int i = 0 ; i < n ; i ++ ){
            ok = 0;
            for(int j = 1 ; j <= m ; j ++ ){
                has[j] = has[j-1] * 131 + s[i][j-1];
            }
            for(int j = 1 ; j <= m ; j ++ ){
                if(mp.count(get(1,j-1) * 131 + get(j,m))){
                    ok = 1;
                    break;
                }
            }
            if(ok == 0)break;
        }
        if(ok)return cout << 2,0;
    }

    int pos1,pos2;
    for(int i = 0 ; i < m ; i ++ ){
        pos1 = i;
        if(s[0][i] != s[1][i])break;
    }
    for(int i = m-1 ; i > 0 ; i -- ){
        pos2 = i;
        if(s[0][i] != s[1][i])break;
    }
    ok = 0;
    for(int i = pos1 ; i < pos2 ; i ++ ){
        if(s[0][i] != s[1][i+1])ok = 1;
    }
    string x;
    for(int i = 0 ; i < pos1 ; i ++ ){
        x += s[0][i];
    }
    string y = x;
    // if(!ok){
        x += s[1][pos1];
        for(int i = pos1 ; i < s[0].size() ; i ++ )x += s[0][i];
    // }
    // else{
        y += s[0][pos1];
        for(int i = pos1 ; i < s[1].size() ; i ++ )y += s[1][i];
    // }
    mp.clear();
    for(int i = 1 ; i <= m+1 ; i ++ ){
        has[i] = has[i-1] * 131 + x[i-1];
    }
    for(int i = 1 ; i <= m+1 ; i ++ ){
        mp[get(1,i-1) * 131 + get(i+1,m+1)];
    }
    ok = 0;
    mp[get(1,m)] = mp[get(2,m+1)] = 1;
    for(int i = 0 ; i < n ; i ++ ){
        ok = 0;
        for(int j = 1 ; j <= m ; j ++ ){
            has[j] = has[j-1] * 131 + s[i][j-1];
        }
        for(int j = 1 ; j <= m ; j ++ ){
            if(mp.count(get(1,j-1) * 131 + get(j,m))){
                ok = 1;
                break;
            }
        }
        if(ok == 0)break;
    }
    int res = ok;
    ok = 0;
    x = y;
    // cout << x << '\n';
    mp.clear();
    for(int i = 1 ; i <= m+1 ; i ++ ){
        has[i] = has[i-1] * 131 + x[i-1];
    }
    for(int i = 1 ; i <= m+1 ; i ++ ){
        mp[get(1,i-1) * 131 + get(i+1,m+1)];
    }
    ok = 0;
    mp[get(1,m)] = mp[get(2,m+1)] = 1;
    for(int i = 0 ; i < n ; i ++ ){
        ok = 0;
        for(int j = 1 ; j <= m ; j ++ ){
            has[j] = has[j-1] * 131 + s[i][j-1];
        }
        for(int j = 1 ; j <= m ; j ++ ){
            if(mp.count(get(1,j-1) * 131 + get(j,m))){
                ok = 1;
                break;
            }
        }
        if(ok == 0)break;
    }
    res += ok;
    cout << res << '\n';
}
0