带分数(嵌套dfs)

原题链接

#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)

原式 n = a +\frac{b}{c} 可以化简至 nc = ac + b

算法流程:利用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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

linengcs

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值