描述
小白手上有一些小木棍,它们长短不一,小白想用这些木棍拼出一个等边三角形,并且每根木棍都要用到。 例如,长度为 1,2,3,3的 4 根木棍,可以让长度为 1,2的木棍组成一条边,另外 2根分别组成 2 条边,拼成一个边长为 3的等边三角形。请问小白用这些木棍是否能够拼出一个等边三角形?
输入
首先输入一个整数 n(3≤n≤20),表示木棍数量,接下来输入 n 根木棍的长度pi(1≤pi≤10000)。
输出
如果小白能拼出等边三角形,输出yes
,否则输出no
。
所以,我们从题目能看出这是一道dfs的问题。而且,我们知道三角形是有三条边的,因此,这道题的时间复杂度为:O(n³)。
来看样例:
代码:
我提交在OJ上之后,就会发现这个程序TLE了。
如果样例是:13 1 1 1 1 1 1 1 1 1 1 1 1 1
就会递归2391484次。
优化方法:剪枝
可能大家会问:什么是剪枝?
简单来说,就是下图:
打叉的就是无用的搜索!
我们就要加上苛刻的条件,让执行dfs比较难
优化代码:
优化后次数:110251
在这里面,用上了if条件句,就不会让dfs乱执行了!
新手发表,有问题请指出,谢谢!
2023.10.4下午16:20