Restoring the Expression

原题：http://codeforces.com/problemset/problem/898/F

题目大意是给你一串数字，让你在数字间添加“+”和“=”使式子成立，输出任意一种方案。其实就是a+b=c的形式。

思路:因为是加法，所以最多进一位。可知a,b,c的长度之间的关系有4种可能，分别是len(a)=len©,len(a)+1=len©,len(b)=len©,len(b)+1=len©。对字符串hash时取base为10，这样任意在原串中截一段得到的子串的hash值就会跟其数字形式的值相等，就可以直接利用(hash(a)+hash(b))%mod==hash©来判断是否成立。

注意：当a,b,c是个位数时，可以以0开头，例如0+0=0

#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1e6+100,base=10,mod=1e9+7;
char s[maxn];
int len;
ll hash[maxn],pw[maxn];
ll gethash(int l,int r);
int check(int al,int bl);
void print(int al,int bl);
int main(){
	scanf("%s",s+1);
	len=strlen(s+1);
	pw[0]=1;
	for (int i=1;i<=len;i++){
		hash[i]=(hash[i-1]*base+s[i]-'0')%mod;
		pw[i]=(pw[i-1]*base)%mod;
	}
	for (int i=1;i<=len/2+1;i++){        //i为与c相等或差1的a或b
        if(check(i,len-2*i)) break;      //此时i为a，al=cl
        if(check(i,len-2*i-1)) break;    //此时i为a，al+1=cl
        if(check(len-2*i,i)) break;      //此时i为b，bl=cl
        if(check(len-2*i-1,i)) break;    //此时i为b，bl+1=cl
	}
	return 0;
}
ll gethash(int l,int r){
	ll res=(hash[r]-hash[l-1]*pw[r-l+1])%mod;  //别这样写(hash[r]-hash[l-1]*pw[r-l+1]+mod)%mod                    
	if (res<0) res+=mod;                       //一个mod可能不足以使负变正
	return res;
}
void print(int al,int bl){
	for (int i=1;i<=al;i++) printf("%c",s[i]);
	printf("+");
	for (int i=al+1;i<=al+bl;i++) printf("%c",s[i]);
	printf("=");
	for (int i=al+bl+1;i<=len;i++) printf("%c",s[i]);
}
int check(int al,int bl){
	if (al<=0||bl<=0||al>len-al-bl||bl>len-al-bl) return 0;      //注意c要是最大的
	if (s[1]=='0'&&al!=1) return 0;
	if (s[al+1]=='0'&&bl!=1) return 0;
	if (s[al+bl+1]=='0'&&len-al-bl!=1) return 0;
	ll a=gethash(1,al),b=gethash(al+1,al+bl),c=gethash(al+bl+1,len);
	if ((a+b)%mod==c){
		print(al,bl);
		return 1;
	}
	return 0;
}