题目描述
给定一个只由0,1,2,3,4,5,6,7,8,9组成的字符串,要求你在字符之间添加"+“或者”-",例如给定的字符串是"12345",你可以找到一个表达式"123+4-5",现在给一个整数N,请找出所有的表达式使得表达式的值为N。两个相邻的字符之间最多只能使用一个符号。
输入格式
输入包含多组测试数据,每组测试数据占一行,是一个字符串和一个整数N。
其中字符串的长度最大为12,N的最大值为999999999999
输出格式
对于每组测试数据,输出能够满足条件的表达式的个数。
输入输出样列
输入样例1:
123456789 3
21 1
输出样例1:
18
1
【耗时限制】1000ms 【内存限制】128MB
代码如下
#include<bits/stdc++.h>
using namespace std;
const char op[3]={'+','-',' '};
char a[15];
int n,b[15],k,ans;
string s;
bool check(){
int sum=0,num=b[1];
char lop='+';
for(int i=1;i<n;i++){
if(a[i]==' ') num=num*10+(b[i+1]);
else{
if(lop=='+') sum=sum+num;
else sum-=num;
lop=a[i];
num=b[i+1];
}
}
if(lop=='+') sum+=num;
else sum-=num;
if(sum==k) return true;
else return false;
}
void dfs(int cur){
if(cur==n){
if(check()) ans++;
return;
}
for(int i=0;i<3;i++){
a[cur]=op[i];
dfs(cur+1);
}
}
int main(){
while(cin>>s>>k){
ans=0;
n=s.size();
for(int i=0;i<n;i++){
b[i+1]=s[i]-'0';
}
dfs(1);
cout<<ans<<endl;
}
return 0;
}