P1149 火柴棒等式
骚粉色了解一下。
参考:https://www.luogu.org/problemnew/solution/P1149?page=2
题意:就是利用火柴棒按照数码上的数字表示,拼凑出来的数字,保证为一个等式。
给你n个火柴棒,问刚好用完并且拼成等式多少种。
做法:思想很简单,题目给了最大的n为24,考虑一下数位越多数字越大,数字最大为:0+x=x =>6+2+x+2+x=24 => x=7;
7能拼出最大的数:711。
这里考虑到:可能另一数为0会影响到这边用到的火柴棒较少,所以:1+x-1=x => 2+2+(x-1)+2+(x)=24 =>
x需要的火柴数和x-1需要的火柴数加起来为18。前面算出来711,这里考虑一下四位数。
1111 1110=20
1112 1111=19
这两种是最少的情况了,所以可以知道最大上限绝对不会超过1111。
PS:事实上1112和1111都是不合法的,我们通过不合法的方法求出来不可能超过的上限,真正上限就应该是上面的711,但是只能证明必要性,我不会证明充分性。
枚举0-1111即可。复杂度为O(n^2),10e6级别的,可以接受。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<stack>
using namespace std;
int E[50];
int F[50];
int cost[10]={6,2,5,5,4,5,6,3,7,6};
int select(int sum)
{
if(sum==0)return 6;
int n=sum;
int digit;
int s=0;
while(n)
{
digit=n%10;
s+=cost[digit];
n/=10;
}
return s;
}
int main()
{
int n;
int ans=0;
cin>>n;
for(int i=0;i<=1000;i++)
for(int j=0;j<=1000;j++)
{
if(select(i)+select(j)+select(i+j)+4==n)
ans++;
}
cout<<ans<<endl;
}
这里面算火柴棒算重复了。
预处理一下效率更高。
#include <iostream>
using namespace std;
int x[1111] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6};
int n;
int ans=0;
int main()
{
int i;
cin>>n;
for(i=10; i<=1110;i++)
x[i]=x[i/10]+x[i%10];
for(int i=0;i<1111;i++)
for(int j=i;j<1111;j++)
if(x[i]+x[j]+x[j-i]+4==n)
ans++;
cout<<ans<<endl;
}
这是我发现,预处理前1111位数字的火柴棒使用数量最有效的方法:通过递归的方式。
每一个两位以上数字可以被分成最后一位和剩余位,而剩余位通过递归在之前已经 求过。
这样的递归方式能在线性时间内递归出来所有数字,而上面的代码是O(n*4),数字越大,对复杂度影响越大。
而且枚举加数和合也可以保证自己的答案少了一半的枚举过程。
这里还介绍一下参考大佬的搜索代码:
递归出所有的火柴棒使用,然后枚举数字搜索。
#include <iostream>
using namespace std;
int x[1001] = {6, 2, 5, 5, 4, 5, 6, 3, 7, 6}, b[4];//初始定义0~9火柴棒个数, b数组存放每次可能的等式
int n, tot = 0;
void search(int l)//搜索
{
int i;
for(i = 0; i <= 999; ++i)
{
if(n - x[i] >= 0)//火柴棍还够
{
b[l] = i;//火柴棒数放入数组
n = n - x[i];//在总数中减去火柴棒
if(l == 3)
{
if(b[1] + b[2] == b[3] && n == 0) tot++;//满足等式且火柴棒用完
}
else search(l + 1);//回溯
n = n + x[i];//保存之前状态
}
}
}
int main()
{
int i;
cin >> n;
n = n - 4;
for(i = 10; i <= 999; ++i)
x[i] = x[i / 10] + x[i % 10];//把2位数火柴棒, 3位数火柴棒放入数组, 理论上可达到11111111,但因为数据弱四位就过了
search(1);
cout << tot;
return 0;
}
tql