快速加法(C++)[DFS]

蒟蒻尹

已于 2022-07-25 22:07:08 修改

阅读量800

点赞数 1

分类专栏： “难题” 文章标签：加法运算最小加号深度优先搜索剪枝算法数字和

于 2022-07-25 22:01:20 首次发布

本文链接：https://blog.csdn.net/ytk888/article/details/125982850

版权

“难题” 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

题目：

Description

小小秦是小秦的弟弟，他才学会加法。

小秦现在想考试一下弟弟对加法的掌握程度，但他又不想出平时学校老师出的那些题

于是他给出一串数字S，再给出一个整数N。

问弟弟这样一个问题在S中添加几个加号，可以使得表达式的结果为S。

例如 S="303",N=6 则只最少只要一个加号就可以了即3+03=6

S="1110",N=3 则最少要加3个加号，即1+1+1+0=3

Format

Input

第1行：1个字符串S(1 ≤ S的长度 ≤ 40) 和1个整数N(N≤100000)。

S和N用1个空格分隔。

Output

第1行：1个整数K，表示最少的加法次数让S等于N。

如果怎么做都不能让S等于N，则输出-1。

Samples

【输入样例】

输入数据 1

99999 45

输出数据 1

思路：

这道题采用 $DFS$ （深度优先）算法。

我们可以模拟添加加号的位置：

用 $Dep$ 指向当前数，

记下当前的累加和 $Sum$ （ $Sum$ 并不一定加到 $Dep$ 的深度），

再记下当前的加数 $NowNum$

和加号的个数 $Jia$ （拼音大法）

注意，关于 $NowNum$ 有两种操作：

$1.$ $NowNum$ 是会延长的，它可以与当前 $Dep$ 指向的数合并，但不会累加，不然会很麻烦，因为有 $Sum$ 记下累加和就够了。如：当前 $NowNum$ $=99$ ，当前 $Dep$ 指向数字 $9$ ，则 $NowNum$ 可以与 $1$ 合并为 $999$ ；

$2.$ $NowNum$ 加到 $Sum$ 里更新累加和，此时 $NowNum$ 变成 $Dep$ 指向的数,此时加号个数也要加一

我们做就可以按这两种操作来做 $01DFS$

$TLE$ 代码：

#include<bits/stdc++.h>
using namespace std;
const int N= 210;
string s;
int len;
int m;
int ans=INT_MAX;//答案 
int zhuan(char ss) {//转换器，将字符型转成整型 
	if(ss=='0') return 0;
	if(ss=='1') return 1;
	if(ss=='2') return 2;
	if(ss=='3') return 3;
	if(ss=='4') return 4;
	if(ss=='5') return 5;
	if(ss=='6') return 6;
	if(ss=='7') return 7;
	if(ss=='8') return 8;
	if(ss=='9') return 9;
}
void dfs(int dep,int sum,int jia,int now_num) {
	if(sum>m) return ;//如果sum比要求的和m还大，可以return了 
	if(dep==len) {//如果已经遍历完这串数 
		if(sum+now_num==m) //且sum为要求的和m 
		    ans=min(jia,ans);//更新答案 
		return ;//一定要return 
	}
// 01DFS: 
	dfs(dep+1,sum,jia,now_num*10+zhuan(s[dep]));
	//now_num它可以与当前dep指向的数合并
	dfs(dep+1,sum+now_num,jia+1,zhuan(s[dep]));
	//也可以把now_num加到sum中，此时jia+1 
}
int main() {
	cin>>s>>m;
	len=s.size();//获取长度 
	dfs(0,0,0,0); 
	if(ans==INT_MAX) puts("-1");
	else
		cout<<ans<<endl;
	return 0;
}

对，你没有看错， $TLE$ 了，这说明没完

于是我又加了几个剪枝：

正解：

#include<bits/stdc++.h>
using namespace std;
const int N = 210, M = 100010;
string s;
int len;
int m;
int ans = INT_MAX;
int bs[N];
/*  可行性剪枝 
    bs[i]代表把数字串中每一个数单独加起来（不合并）加到第i位 （类似前缀和） 
    此是1~i贡献的值最小 
    如一列数1、2、3、4、5 
    则bs[3]=1+2+3+4=10;  */
int ss[M]; 
/*  最优性剪枝
    ss[i]代表凑出i要花费的最少的加号数量(i是累加和，相当于DFS中的sum) */ 
int zhuan (char ss)
{
	if (ss == '0') return 0;
	if (ss == '1') return 1;
	if (ss == '2') return 2;
	if (ss == '3') return 3;
	if (ss == '4') return 4;
	if (ss == '5') return 5;
	if (ss == '6') return 6;
	if (ss == '7') return 7;
	if (ss == '8') return 8;
	if (ss == '9') return 9;
}
void dfs (int dep, int sum, int jia, int now_num)
{
	if (sum > m) return ;
	if (now_num + sum > m) return ;
	if (dep == len)
	{
		if (sum + now_num == m) ans = min (jia, ans);
		return ;
	}

	if (sum + bs[len - 1] - bs[dep] > m) return ;
	//如果把dep到末位的数一个个加起来都大于m的话就return吧 
	if (ss[sum])//如果不是最优的，return； 
		if (ss[sum] < jia) return ;
	ss[sum] = jia;//否则更新最优的 
	
	dfs (dep + 1, sum, jia, now_num * 10 + zhuan (s[dep]) );
	dfs (dep + 1, sum + now_num, jia + 1, zhuan (s[dep]) );
}
int main()
{
	cin >> s >> m;
	len = s.size();
	bs[0] = zhuan (s[0]);
	for (int i = 1; i < len; i++)//做前缀和 
		bs[i] = bs[i - 1] + zhuan (s[i]);
	dfs (0, 0, 0, 0);
	if (ans == INT_MAX) puts ("-1");
	else
		cout << ans << endl;
	return 0;
}