码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 898F 字符串hash

时间:2018-02-19 12:41:29      阅读:181      评论:0      收藏:0      [点我收藏+]

标签:快速   ssi   first   pos   one   技术分享   turn   --   检验   

F. Restoring the Expression

题意:给出一个字符串,要把它折分成三部分 a、b、c , 使得 a+b=c 。输出任何一种可行情况。

tags:字符串 hash

因为 a+b=c ,所以 lena、lenb 至少要有一个等于 lenc 或 lenc-1  。所以枚举 lenc,每次检验一下可不可行。

但每次都暴力检验肯定超时,这里把字符串hash 一下,根据 hash值 快速检验。

 

记一下模板

const int mod = 1e9+7;
ll  Hash[N], p[N], Base=10;
void Init(string si)
{
    int len = si.size();
    Hash[0]=0;
    for(int i=0; i<len; ++i)
        Hash[i+1] = (Hash[i]*Base%mod+si[i]-0)%mod;
    p[0]=1;
    for(int i=0; i<len; ++i)
        p[i+1] = p[i]*Base%mod;
}
ll  get(int l, int r)
{
    if(r<0 || l-1<0 || r-l+1<0) return 0;
    return (Hash[r]-Hash[l-1]*p[r-l+1]%mod+mod)%mod;
}
技术分享图片
#include<bits/stdc++.h>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define rep(i,a,b) for (int i=a; i<=b; ++i)
#define per(i,b,a) for (int i=b; i>=a; --i)
#define mes(a,b)  memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define MP make_pair
#define PB push_back
#define fi  first
#define se  second
typedef long long ll;
const int N = 1000005;

const int mod = 1e9+7;
ll  Hash[N], p[N], Base=10;
void Init(string si)
{
    int len = si.size();
    Hash[0]=0;
    for(int i=0; i<len; ++i)
        Hash[i+1] = (Hash[i]*Base%mod+si[i]-0)%mod;
    p[0]=1;
    for(int i=0; i<len; ++i)
        p[i+1] = p[i]*Base%mod;
}
ll  get(int l, int r)
{
    if(r<0 || l-1<0 || r-l+1<0) return 0;
    return (Hash[r]-Hash[l-1]*p[r-l+1]%mod+mod)%mod;
}

string si;
int len, lenc;
void print(int lena, int lenb)
{
    rep(i,0,lena-1) putchar(si[i]);  putchar(+);
    rep(i,lena,lena+lenb-1) putchar(si[i]); putchar(=);
    rep(i,lena+lenb,len-1) putchar(si[i]);
}

bool check(int lena, int lenb)
{
    string s1=si.substr(0,lena);
    string s2=si.substr(lena,lenb);
    string s3=si.substr(lena+lenb,len-lena-lenb);
    int flag=0, len3=len-lena-lenb, i, j;
    for(i=lena-1, j=lenb-1; i>=0 && j>=0; --i, --j)
    {
        int a1=s1[i]-0, a2=s2[j]-0, a3=a1+a2+flag;
        if(a3>=10) a3-=10, flag=1;
        else flag=0;
        if(a3!=s3[len3-(lena-i)]-0) return false;
    }
    return true;
}
bool solve(int lena, int lenb)
{
    if(lena>lenc || lenb>lenc) return false;
    if(lena<0 || lenb<0) return false;
    if(si[lena]==0 && lenb!=1) return false;
    if(si[lena+lenb]==0 && len-lena-lenb!=1) return false;
    if( (get(1,lena)+get(lena+1, lena+lenb))%mod != get(lena+lenb+1, len) ) return false;
    //return true;
    return  check(lena, lenb);
}
int main()
{
    cin>>si;
    Init(si);
    len=si.size();
    for(lenc=1; lenc<=len-2; ++lenc)
    {
        if(solve(lenc, len-lenc*2)) {
            print(lenc, len-lenc*2); return 0;
        }
        if(solve(lenc-1, len-lenc-(lenc-1))) {
            print(lenc-1, len-lenc-(lenc-1)); return 0;
        }
        if(solve(len-lenc*2, lenc)) {
            print(len-lenc*2, lenc); return 0;
        }
        if(solve(len-lenc-(lenc-1), lenc-1)) {
            print(len-lenc-(lenc-1), lenc-1); return 0;
        }
    }

    return 0;
}
View Code

Codeforces 898F 字符串hash

标签:快速   ssi   first   pos   one   技术分享   turn   --   检验   

原文地址:https://www.cnblogs.com/sbfhy/p/8433653.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!