HDU 4403 A very hard Aoshu problem （DFS暴力）

原创于 2014-07-07 20:53:34 发布 · 688 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

dfs 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种算法，该算法能够针对给定的数字字符串，在适当位置插入运算符‘+’和等于符号‘=’，使得字符串两侧的数值表达式计算结果相等。通过递归深度优先搜索的方法实现，考虑所有可能的分割方式。

题意：给你一个数字字符串，问在字符串中间加‘=’、‘+’使得‘=’左右两边相等。

1212 ： 1+2=1+2， 12=12；

12345666 ： 12+3+45+6=66， 1+2+3+4+56=66；

#include<stdio.h>
#include<string.h>
#include<string>
#include<map>
#include<stack>
#include<math.h>
#include<queue>
#include<vector>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define M 20005
vector<int>Ga,Gb;
char s[50];
int l,len,ans;
int a[M],b[M];
int num[55][55];
int get_num(int x,int y)
{
	int sum=0;
	for(int i=x;i<=y;i++)
		sum=sum*10+s[i]-'0';
	return sum;
}
void dfs2(int i,int lsum,int rsum)
{
	if(i>=len)
	{
		if(lsum==rsum)
			ans++;
		return;
	}
	for(int k=i;k<len;k++)
		dfs2(k+1,lsum,rsum+num[i][k]);
}
void dfs1(int i,int sum)
{
	if(i>=l)
		dfs2(i,sum,0);
	for(int k=i;k<l;k++)
		dfs1(k+1,sum+num[i][k]);
}
int main()
{
	while(scanf("%s",s))
	{
		if(s[0]=='E')
			break;
		ans=0;
		int i,j,k;
		len=strlen(s);
		for(i=0;i<len;i++)
		{
			for(j=i;j<len;j++)
			{
				num[i][j]=get_num(i,j);
			}
		}
		for(int a=1;a<len;a++)
		{
			l=a;
			dfs1(0,0);
		}
		printf("%d\n",ans);	
	}
	return 0;
}
/*
1212
12345666
1235
*/