P1618 三连击（升级版）

题目描述

将 1,2,…,9 共 9 个数分成三组，分别组成三个三位数，且使这三个三位数的比例是 A:B:C，试求出所有满足条件的三个三位数，若无解，输出 No!!!。

//感谢黄小U饮品完善题意

输入格式

三个数，A,B,C。

输出格式

若干行，每行 3 个数字。按照每行第一个数字升序排列。

输入输出样例

输入 #1复制

1 2 3

输出 #1复制

192 384 576
219 438 657
273 546 819
327 654 981

说明/提示

保证 0≤A<B<C≤999。

upd 2022.8.3：新增加二组 Hack 数据。
代码：
一开始我使用的是深搜：

#include <bits/stdc++.h>
using namespace std;

int A, B, C;
bool found = false;

void dfs(int pos, unsigned used, int x, int y, int z) {
    if (pos == 9) {
        if (x % A == 0 && y % B == 0 && z % C == 0) {
            int k = x / A;
            if (y / B == k && z / C == k) {
                cout << x << " " << y << " " << z << "\n";
                found = true;
            }
        }
        return;
    }

    // 分阶段生成x, y, z的数字
    if (pos < 3) {
        for (int i = 1; i <= 9; ++i) {
            if (!(used & (1 << i))) {
                dfs(pos + 1, used | (1 << i), x * 10 + i, y, z);
            }
        }
    } else if (pos < 6) {
        for (int i = 1; i <= 9; ++i) {
            if (!(used & (1 << i))) {
                dfs(pos + 1, used | (1 << i), x, y * 10 + i, z);
            }
        }
    } else {
        for (int i = 1; i <= 9; ++i) {
            if (!(used & (1 << i))) {
                dfs(pos + 1, used | (1 << i), x, y, z * 10 + i);
            }
        }
    }
}

int main() {
    cin >> A >> B >> C;
    dfs(0, 0, 0, 0, 0);
    if (!found) {
        cout << "No!!!\n";
    }
    return 0;
}

 但是这样还有一组测试用例超时，所以后面又写了一个代码：

#include <bits/stdc++.h>
using namespace std;

bool check(int x, int y, int z) {
    bool digits[10] = {false}; // 初始化所有数字未出现（digits[0]~digits[9]）
    
    // 检查单个数字的函数
    auto check_num = [&digits](int num) {
        while (num > 0) {
            int d = num % 10; // 取出最后一位数字
            if (d == 0 || digits[d]) { // 如果数字是0或已经出现过
                return false;
            }
            digits[d] = true; // 标记该数字已出现
            num /= 10;       // 去掉最后一位
        }
        return true;
    };

    // 检查三个数字
    if (!check_num(x) || !check_num(y) || !check_num(z)) {
        return false;
    }

    // 检查1-9是否都出现了
    for (int i = 1; i <= 9; ++i) {
        if (!digits[i]) {
            return false;
        }
    }
    return true;
}

int main() {

    int A, B, C;
    cin >> A >> B >> C;
    bool found = false;
    for (int k = 1; ; ++k) {
        int x = k * A;
        int y = k * B;
        int z = k * C;
        if (x > 999 || y > 999 || z > 999) break;
        if (x < 100 || y < 100 || z < 100) continue;
        if (check(x, y, z)) {
            cout << x << " " << y << " " << z << "\n";
            found = true;
        }
    }
    if (!found) {
        cout << "No!!!\n";
    }
    return 0;
}

时间效率更高