acwing 带分数 y总优化代码

资源限制
时间限制：1.0s 内存限制：256.0MB

问题描述
100 可以表示为带分数的形式：100 = 3 + 69258 / 714。

还可以表示为：100 = 82 + 3546 / 197。

注意特征：带分数中，数字1~9分别出现且只出现一次（不包含0）。

类似这样的带分数，100 有 11 种表示法。

输入格式
从标准输入读入一个正整数N (N<1000*1000)

输出格式
程序输出该数字用数码1~9不重复不遗漏地组成带分数表示的全部种数。

注意：不要求输出每个表示，只统计有多少表示法！

样例输入1
100

样例输出1
11

样例输入2
105

样例输出2
6

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 10;

int n;
bool st[N], backup[N];//判重数组
int ans;

bool check(int a, int c)//主要是判断 b合不合格
{
    long long b = n * (long long)c - a * c;

    if (!a || !b || !c) return false; //都不能为0

    memcpy(backup, st, sizeof st); //为了不影响之后dfs操作 需要复制一份当前st值用来检验b
    while (b)
    {
        int x = b % 10;     // 取个位
        b /= 10;    // 个位删掉
        if (!x || backup[x]) return false; // x为0 ，或者之前出现过。检查是否存在已经用过的
        backup[x] = true;//标记一下，已经用过
    }

    for (int i = 1; i <= 9; i ++ )
        if (!backup[i])
            return false;/检查是否存在没有用的

    return true;
}

void dfs_c(int u, int a, int c)
{
    if (u > n) return;

    if (check(a, c)) ans ++ ;//检查a，c是不是合格；

    for (int i = 1; i <= 9; i ++ )
        if (!st[i])
        {
            st[i] = true;
            dfs_c(u + 1, a, c * 10 + i);//在一个数后面加上一位，需要×10
            st[i] = false;//恢复现场
        }
}

void dfs_a(int u, int a)
{
    if (a >= n) return;
    if (a) dfs_c(u, a, 0);

    for (int i = 1; i <= 9; i ++ )
        if (!st[i])
        {
            st[i] = true;
            dfs_a(u + 1, a * 10 + i);
            st[i] = false;
        }
}

int main()
{
    cin >> n;

    dfs_a(0, 0);

    cout << ans << endl;//已经用了几个数字，第二个是a的

    return 0;
}