Vasya and Golden Ticket-Codeforce 512C-前缀和

题目传送门
题意：
给你一串数字，判断是否能将这串数字分成和相等的几部分。
思路：
首先将0剔除掉，因为0没有什么影响，只有两种特殊情况有影响，提前判断一下即可。然后用前缀和算出所有可能区间的和，并且记录区间的左右端点，然后排序。遍历所有的区间，找到左端点为1的区间，然后在值相等的区间中找到可以连接的区间，知道区间的右端点等于终点。复杂度就是O（5050）,O(n^2/2).
虽然没有某些大佬们写的简洁，但也是我努力的成果。另外自己考虑问题不够全面，思维还不行，需要多加练习。
官方题解做法在后面。
具体看代码注释。

AC code:
#include<iostream>
#include<cstdio>
#include<vector>
#include<bitset>
#include<stack>
#include<set>
#include<queue>
#include<map>
#include<cmath>
#include<string>
#include<cstring>
#include<ctime>
#include<fstream>
#include<cstdlib>
#include<algorithm>

using namespace std;

#define pii pair<int, int>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define per(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a,b) for(int i=a;i>=b;i--)
#define all(x) x.begin(),x.end()
#define PER(i,x) for(auto i=x.begin();i!=x.end();i++)
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
typedef long long LL;
const double eps=1.0e-5;
const int maxn=100+ 10;
int dx[4] = {0,0,-1,1};
int dy[4] = {1,-1,0,0};
int n = 0,size = 0; 
int a[maxn];
int pre_sum[maxn];
char str[maxn]; 
struct node{
	int val;
	int s,e;
};
node nod[maxn*maxn];
bool cmp1(node a,node b){
	if(a.val != b.val){
		return a.val < b.val;
	}else{
		return a.s < b.s;
	}
}
void solve(){
	int cnt = 0;
	per(i,0,size-1){//计算区间和 
		per(j,i+1,size){
			nod[++cnt].val = pre_sum[j] - pre_sum[i];
			nod[cnt].s = i + 1;nod[cnt].e = j;
		}
	}
	nod[cnt+1].val = 1e3; 
	sort(nod + 1,nod + cnt + 1,cmp1);//排序 
	for(int i = 1;i <= cnt;){
		if(nod[i].s != 1){//找到开始区间为1的 
			i++;
			continue;
		}
		int j = i + 1;
		bool flag = false;
		while(nod[j].val == nod[i].val){//如果等值的区间只有一个，直接continue 
			flag = true;
			j++;
		}
		if(flag != true){
			i = j;
			continue;
		}
		int temp = i;
		for(int k = i+1;k <= j-1;++k){
			while(nod[k].s != nod[temp].e + 1 && k <= j - 1){
			//找到下一个等值并且区间相接的区间（因为等值的区间可能会重叠，所以并不是检查
			//相邻的区间，99那组测试样例就是这样WA的 
				k++;
			}
			if(k > j-1){
				i = j;
				break;
			}
			temp = k;
			if(nod[temp].e == size){
				printf("YES\n");
				return ;
			}
		}
		i = j;//记得赋值，否则陷入死循环 
	}
	printf("NO\n");
}
int main(){
	#ifndef ONLINE_JUDGE
		freopen("a.txt","r",stdin);
	#endif
	while(~scanf("%d",&n)){
		mem(pre_sum,0);
		scanf("%s",str);
		size = 0;//记录非0数字的数量 
		int ct = 0;
		per(i,0,n-1){
			if(str[i] - '0' == 0){//去掉0 
				ct++;
				continue;
			}
			a[++size] = str[i] - '0';
			pre_sum[size] = pre_sum[size-1] + a[size];//计算前缀和 
		}
		if(ct == n){//0的特殊情况处理 
			printf("YES\n");
			continue;
		}
		if(ct == n-1){
			printf("NO\n");
			continue;
		}
		solve();
	}
	
	return 0; 
}

官方题解做法：

AC code:
#include<iostream>
#include<cstdio>
#include<vector>
#include<bitset>
#include<stack>
#include<set>
#include<queue>
#include<map>
#include<cmath>
#include<string>
#include<cstring>
#include<ctime>
#include<fstream>
#include<cstdlib>
#include<algorithm>

using namespace std;

#define pii pair<int, int>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define per(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a,b) for(int i=a;i>=b;i--)
#define all(x) x.begin(),x.end()
#define PER(i,x) for(auto i=x.begin();i!=x.end();i++)
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
typedef long long LL;
const double eps=1.0e-5;
const int maxn=100 + 10;
int dx[4] = {0,0,-1,1};
int dy[4] = {1,-1,0,0};
int n = 0; 
char str[maxn];
void solve(){
	per(i,0,n-2){
		int sum1 = 0;
	 	per(j,0,i){
	 	 	sum1 += str[j] - '0'; 
		}
		int k = i + 1;
		while(k < n){
			int sum2 = str[k++] - '0';//不能用0作为初始值，否则与下面的while循环的判断条件
			//有冲突，会陷入死循环，比如现在k->3,sum1->2,k+1->1,如果初始化为0,k++无法进行就会
			//死循环 
			//int sum2 = 0;
		  	while(k < n && sum2  + str[k] - '0'<= sum1){
		  //while(k < n && sum2 < sum1){//这种判断条件可能会导致sum2>sum1 
		  		sum2 += str[k++] - '0';
			}
			if(sum2 != sum1){
				break;
			}
			if(sum2 == sum1 && k == n ){//将隐含条件最好写上，防止出现 
				printf("YES\n");
				return ;
			}
		}
		
	}
	printf("NO\n");
}
int main(){
	#ifndef ONLINE_JUDGE
		//freopen("a.txt","r",stdin);
	#endif
	while(~scanf("%d",&n)){
		scanf("%s",str);
		solve();
	}
	return 0; 
}