【洛谷 P1618】三连击（升级版）题解（深度优先搜索+位集合）

三连击（升级版）

题目描述

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

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

输入格式

三个数，$A,B,C$。

输出格式

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

样例 #1

样例输入 #1

1 2 3

样例输出 #1

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

提示

保证 $A<B<C$。

---

$\text{upd 2022.8.3}$：新增加二组 Hack 数据。

思路

搜索得到全排列，用乘法判断是否符合比例关系。

AC代码

#include <iostream>
#include <bitset>
#include <cmath>
#define AUTHOR "HEX9CF"
using namespace std;

int a, b, c;
int x, y, z;
bitset<15> bs;
int arr[15];
int flg;

void f()
{
    if (9 == bs.count())
    {
        a = 0;
        b = 0;
        c = 0;
        for (int i = 0; i < 3; i++)
        {
            a += arr[i + 1] * pow(10, 2 - i);
            b += arr[i + 4] * pow(10, 2 - i);
            c += arr[i + 7] * pow(10, 2 - i);
        }
        if ((a * y == b * x) && (a * z == c * x) && (b * z == c * y))
        {
            cout << a << " " << b << " " << c << endl;
            flg = 1;
        }
        return;
    }
    for (int i = 1; i <= 9; i++)
    {
        if (!bs[i])
        {
            bs[i] = 1;
            arr[bs.count()] = i;
            f();
            bs[i] = 0;
        }
    }
}

int main()
{
    bs.reset();
    cin >> x >> y >> z;
    f();
    if(!flg){
        cout << "No!!!" << endl;
    }
    return 0;
}