补题:Div2 #578 E. Compress Words

补题:Div2 #578 E. Compress Words

从第一个单词开始,如果前一个单词的后缀和后一个单词的前缀相同,那么把他们合并,求最后合并后的串

每新读入一个新字符串就将其和已有的字符串合并,不是合并i和i-1

想法就是对想合并的两个串求哈希值,然后枚举能合并的前缀后缀的长度

单模数hash在codeforces上很容易被hack的,建议双模数hash

如果时间超限的话,可以考虑预处理快速幂,能少个log

(hash查询是O(1)的,所以在某些情况下可以往稍稍暴力的方面想)

(太久没写被卡细节=_=)

#include<bits/stdc++.h>
using namespace std;
#define PI acos(-1)
#define ll long long
#define clean(a,b) memset(a,b,sizeof(a))
const int inf=0x3f3f3f3f;
const int maxn=1e6+9;
const int mod=1e9+7;
const int mod_=1e9+9;
inline ll read()
{
    register ll x=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
    return x*f;
}
string s[maxn];
int base=131;
ll _hash[1000009][6];
int fpo[maxn][3];
ll fpow(ll a,ll b,ll c)
{
    a%=c;
    ll ans=1;
    while(b)
    {
        if(b&1) ans=ans*a%c;
        a=a*a%c;
        b>>=1;
    }
    return ans%c;
}

int get_hash(int l,int r,int x)
{
    return (_hash[r+1][x]-_hash[l][x]*fpo[r-l+1][1]%mod+mod)%mod;
}
int get_hash_(int l,int r,int x)
{
    return (_hash[r+1][x]-_hash[l][x]*fpo[r-l+1][2]%mod_+mod_)%mod_;
}

int main()
{
#ifdef LOCAL
    freopen("in.in","r",stdin);
    freopen("out.out","w",stdout);
#endif
    for(int i=1;i<=1e6;i++) fpo[i][1]=fpow(base,i,mod)%mod,fpo[i][2]=fpow(base,i,mod_)%mod_;
    int n=read();
    cin>>s[1];
    int len=s[1].size();
    // int now=2;//now 
    for(int i=0;i<len;i++)
    {
        _hash[i+1][1]=((_hash[i][1]*base)+s[1][i])%mod;
        _hash[i+1][3]=((_hash[i][3]*base)+s[1][i])%mod_;
    }
    string ans=s[1];
    for(int i=2;i<=n;i++)
    {
        cin>>s[i];
        int len=s[i].size();
        int anslen=ans.size();
        int nowlen=min(len,anslen);
        for(int j=0;j<len;j++)
        {
            _hash[j+1][2]=((_hash[j][2]*base)+s[i][j])%mod;
            _hash[j+1][4]=((_hash[j][4]*base)+s[i][j])%mod_;
        }
        int anss=0;
        for(int j=1;j<=nowlen;j++)
        {
            ll hash_1=get_hash(anslen-j,anslen-1,1);
            ll hash_2=get_hash(0,j-1,2);
            ll hash_3=get_hash_(anslen-j,anslen-1,3);
            ll hash_4=get_hash_(0,j-1,4);
            if(hash_1==hash_2) anss=j;
        }
        for(int j=anss;j<len;j++)
        {
            int k=j-anss+anslen;
            _hash[k+1][1]=((_hash[k][1]*base)+s[i][j])%mod;
            _hash[k+1][3]=((_hash[k][3]*base)+s[i][j])%mod_;
            ans+=s[i][j];
        }
    }
    cout<<ans;
    return 0;
}
/*
12345678
678945646

*/

posted @ 2021-07-15 17:22  L·S·D  阅读(43)  评论(0编辑  收藏  举报