題目鏈接:戳這裏
題意:給一個字符串,向裏面添加“+”或“-”,使之結果等於N。
先自己做了一遍,結果不對,最後發現是我考慮不周全,我的搜索過程,只加上或減去了運算符之後的第一個數字,導致最後結果少了一些(見示例)。正確的方法:搜索到每一位數字時,要加上或減去它之後的所有組合情況。
最初的錯誤代碼:
#include<string>
#include<iostream>
using namespace std;
string str;
int len,cnt;
long long n;
void DFS(int x,int sum)
{
if(x==len)
{
if(sum==n)
cnt++;
return;
}
DFS(x+1,sum+(str[x]-'0'));
DFS(x+1,sum-(str[x]-'0'));
DFS(x+1,sum*10+(str[x]-'0'));
}
int main()
{
while(cin>>str>>n)
{
len=str.length();
cnt=0;
DFS(1,(str[0]-'0'));
cout<<cnt<<endl;
}
return 0;
}
AC的代碼:
#include<string>
#include<iostream>
using namespace std;
string str;
int len,cnt;
long long n;
void DFS(int x,int sum)
{
if(x==len)
{
if(sum==n)
cnt++;
return;
}
long long tmp=0;
for(int i=x;i<len;i++)
{
tmp=tmp*10+(str[i]-'0');
DFS(i+1,sum+tmp);
if(x)
DFS(i+1,sum-tmp);
}
}
int main()
{
while(cin>>str>>n)
{
len=str.length();
cnt=0;
DFS(0,0);
cout<<cnt<<endl;
}
return 0;
}