swust oj 2469: 小Y的难题1

原创于 2015-05-08 14:26:17 发布 · 498 阅读

0 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种通过深度优先搜索(DFS)算法解决数学表达式中加减运算与等号匹配的问题，实现了一个递归函数来遍历所有可能的组合，并统计满足条件的等式数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门：http://www.oj.swust.edu.cn/problem/show/2469

应该还有更好的剪枝，不过还没想到。。。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,a[20];
char x[20];
bool flag;
int ans;

void dfs(int k)
{
if(k>n-1) return ;
if(k==n-1&&!flag)
{
int i=0,sum1=a[0],sum2=0;
for(;x[i]!='='&&i<n-1;i++)
{
if(x[i]=='+') sum1+=a[i+1];
else sum1-=a[i+1];
}
for(;i<n-1;i++)
{
if(x[i]=='+') sum2+=a[i+1];
else if(x[i]=='-') sum2-=a[i+1];
else sum2=a[i+1];
}
if(sum2==sum1)
ans++;
return ;
}

x[k]='+';
dfs(k+1);

x[k]='-';
dfs(k+1);

if(flag)
{
flag=false;
x[k]='=';
dfs(k+1);
flag=true;
}
}

int main()
{
while(scanf("%d",&n)!=EOF)
{
flag=true; ans=0;
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0);
printf("%d\n",ans);
}
return 0;
}