CF898F

CF898F

题意

给定一个数字串，要求在串中插入’+‘和’=’，使得这个数字串变成一个成立的等式。要求如下：

1.’+‘必须在’='前面

2.加数与和都不能有前导0，且非空。

输入保证有解。
​
分析：
首先定位=的位置x，这样结果的长度就为len=n-x+1，这样复杂度就变为O(1)。
只剩下被加数长度为len或len-1，或者是加数的长度为len或len-1。
运用双哈希匹配字符串，单哈希会被卡。
注意：
1、每个数都最少一位
2、被加数为0、加数为0、和为0（这坑我跳进去了）的情况

我的代码

#include<iostream>
#include<string>
#include<algorithm>
#include<string.h>
#define ull long long 
#define N 1000005
#define M1 19760817
#define M2 1610612741
#define base 10
using namespace std;
int n;
string s;
ull h1[N],h2[N],b1[N],b2[N];
void gethash(string s){
	b1[0]=b2[0]=1;
	for(int i=1;i<=n;i++){
		h1[i]=(h1[i-1]*base+s[i-1]-'0')%M1;
		h2[i]=(h2[i-1]*base+s[i-1]-'0')%M2;
		b1[i]=(b1[i-1]*base)%M1;
		b2[i]=(b2[i-1]*base)%M2;
	}
}
ull get1(int l,int r){
	return ((h1[r]-h1[l-1]*b1[r-l+1]%M1)%M1+M1)%M1;  		//负数 
}
ull get2(int l,int r){
	return ((h2[r]-h2[l-1]*b2[r-l+1]%M2)%M2+M2)%M2;  		//负数 
}
bool judge(int l,int r){		//l位置前放+，r前放= 
	if(l!=1 &&(get1(1,l-1)+get1(l,r-1))%M1==get1(r,n)&&(get2(1,l-1)+get2(l,r-1))%M2==get2(r,n)){
		if(s[0]=='0'&&l!=2)		//被加数为0 
			return false;
		if(s[l-1]=='0'&&l!=r-1){	//加数为0 
			return false;
		} 
		
		return true;
	} 
	return false;
}
void output(int l,int r){
	for(int i=0;i<l-1;i++)
		cout<<s[i];
	cout<<"+";
	for(int i=l-1;i<r-1;i++)
		cout<<s[i];
	cout<<"=";
	for(int i=r-1;i<n;i++)
		cout<<s[i];
	cout<<endl;
}
int main(){ 
	cin>>s;
	n=s.length(); 
	gethash(s);
	int len;
	for(int i=n;i>(n+1)/2;i--){		//i位置的前面加= 
		len=n-i+1;
		if(judge(len+1,i)){
			output(len+1,i);
			return 0;
		}
		else if(judge(len,i)){
			output(len,i);
			return 0;
		}
		else if(judge(i-len,i)){
			output(i-len,i);
			return 0;
		}
		else if(judge(i-len+1,i)){
			output(i-len+1,i);
			return 0;
		}
	}
}