CF898 F. Restoring the Expression

题目链接
题目大意：
给你长度为n的一个字符串，将其拆分成三部分a,b,c，使其三者相加恰好是一个正确的数学求和式a+b=c

思路：
首先，将整个字符串拆分成n个高精度数，之后运用字符串哈希的方法，将每一个高精度数转换成对应的哈希值，便于求和检查是否相等。
在这之后，遍历c的所有可能长度，由于 a和b的长度必然不会大于c, 可知：前面两式，一个式子长度可能为lenc或lenc-1,另一个就是 n-2 * len2或n-2 * lenc+1 ，在这个结论的基础上，利用式子的哈希值判断它们相加是否满足题目条件，如果满足，则进行输出。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cmath>

#define fast ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define endl "\n"
#define pii pair<int,int>
#define pb push_back
#define debug cout << "visit!" << endl

using namespace std;

const int N = 1e6+5;
const int mod = 1e9+7;
ll Hash[N], p[N], base = 10;
/*Hash[]:用于储存字符串的哈希值*/
void init(string s)
{
	int len = s.size();
	Hash[0] = 0;
	for(int i=0;i<len;++i)
	{
		Hash[i+1] = (Hash[i]*base%mod+s[i]-'0')%mod;/*将一个高精度数压缩成一个Hash值*/
	}
	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;
	}
	else
	{
		return (Hash[r]-Hash[l-1]*p[r-l+1]%mod+mod)%mod;
	}
}

string s;
int len, lenc;
void print(int lena,int lenb)
{
	for(int i=0;i<=lena-1;++i)
	{
		cout << s[i];
	}
	cout << '+';
	for(int i=lena;i<=lena+lenb-1;++i)
	{
		cout << s[i];
	}
	cout << '=';
	for(int i=lena+lenb;i<=len-1;++i)
	{
		cout << s[i];
	}
	cout << endl;
}

bool check(int lena,int lenb)/*检查最终得到的式子是否是一个合法的式子*//*避免哈希值冲突*/
{
	int len3 = len-lena-lenb;
	string s1 = s.substr(0,lena);
	string s2 = s.substr(lena,lenb);
	string s3 = s.substr(lena+lenb,len3);
	int flag = 0, 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(s[lena]=='0'&&lenb!=1)/*以0为首且后面还有数字*/
	{
		return false;
	}
	if(s[lena+lenb]=='0'&&len-lena-lenb!=1)/*以0为首且后面还有数字*/
	{
		return false;
	}
	if((get(1,lena)+get(lena+1,lena+lenb))%mod!=get(lena+lenb+1,len))/*a+b!=c*/
	{
		return false;
	}
	return check(lena,lenb);/*最后进行合法性检查，避免哈希值冲突*/
}

int main()
{
	fast;
	cin >> s;
	init(s);
	len = s.size();
	for(lenc=1;lenc<=len-2;++lenc)/*遍历和的式子长度*/
	{
		if(solve(lenc, len-lenc*2)) /*a和c长度相同*/
		{
            print(lenc, len-lenc*2); return 0;
        }
        if(solve(lenc-1, len-lenc-(lenc-1))) /*a比c短一点*/
		{
            print(lenc-1, len-lenc-(lenc-1)); return 0;
        }
        if(solve(len-lenc*2, lenc)) /*b和c长度相同*/
		{
            print(len-lenc*2, lenc); return 0;
        }
        if(solve(len-lenc-(lenc-1), lenc-1)) /*b比c短一点*/
		{
            print(len-lenc-(lenc-1), lenc-1); return 0;
        }
	}
	return 0;
}

原文请戳这