#1 纯暴力搜索
算法流程:利用next_permutation得到数组1-9的每种排列情况,用暴力双重循环来遍历出介于数组之间的任意两个位置(夹板),从而将数组分成三部分,分别为a,b,c,再判断a.b.c是否符合要求,记录个数
要点:对于计算一个数组中任意位置之间的数字,采用下列代码可以很好解决,l为起点,r为终点
int calu(int l, int r)
{
int res=0;
for(int i=l; i<=r; i++)
res = res*10 + num[i];
return res;
}
另外对于全是数字的数字最好用int数组来存,不要用char或者string来存,简化过程运算
完成AC代码(耗时3362ms)
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn 10
int num[maxn];
int n,a,b,c;
int cnt;
int calu(int l, int r)
{
int res=0;
for(int i=l; i<=r; i++)
res = res*10 + num[i];
return res;
}
int main()
{
cin>>n;
for(int i=1; i<=9; i++) num[i]=i;
do{
for(int i=1; i<=7; i++)
for(int j=i+1; j<=8; j++)
{
a = calu(1,i);
b = calu(i+1,j);
c= calu(j+1,9);
if( n*c == a*c+b)
cnt++;
}
}while(next_permutation(num+1, num+10));
cout<<cnt;
return 0;
}
#2 优化(嵌套dfs)
原式 可以化简至
算法流程:利用dfs来生成1-9的全排列,并在过程中把确定的数字定为a,同时每确定一次a(a不为0)都再进行一次dfs来生成c,每生成一次c都检查一下根据a和c得到的b是否符合要求,是的话则计数,细节问题看代码
完整AC代码(耗时817ms 提高了4倍速度)
#include <iostream>
#include <cstring>
using namespace std;
#define maxn 10
int n,st[maxn],backup[maxn];
int a,b,c;
int cnt;
bool check(int a, int c)
{
// 因为n*c最大可能为10^6 * 10^ 8 = 10^14 会爆int 所以使用long long
long long b = n*(long long)c - a*c;
memcpy(backup, st, sizeof st);
if(!a || !b || !c) return false;
while(b)
{
int x = b%10;
b /= 10;
if(!x || backup[x]) return false; //不能修改全局的状态st,所以要增一个备份
backup[x]=true;
}
//判断是不是1-9都用了
for(int i=1; i<=9; i++)
{
if(!backup[i])
return false;
}
return true;
}
void dfs_c(int cur, int a, int c)
{
if(cur == 9) return ;
if(check(a,c)) cnt++;
for(int i=1; i<=9; i++)
{
if(!st[i])
{
st[i]=true;
dfs_c(cur+1,a,c*10+i);
st[i]=false;
}
}
return;
}
void dfs_a(int cur, int a)
{
if(a>=n) return;
if(a) dfs_c(cur, a, c);
for(int i=1; i<=9; i++)
{
if(!st[i])
{
st[i] = true;
dfs_a(cur+1,a*10+i);
st[i] = false;
}
}
return ;
}
int main()
{
cin>>n;
dfs_a(0,0);
cout << cnt;
return 0;
}