蓝桥杯C++B组决赛---凑平方数

题意

把 0-9 这 10 个数字，不重不漏分成多个组，每个组恰好是一个平方数。问有多少种分组方法？

分析

若要暴力枚举需要分成几组数字，每个组需要分哪些数字，时间复杂度为枚举分组个数 * 枚举每个分组的数字不能接受。换种思路，先确定分组，保证每一组自身都是完全平方数，再考虑把完全平方数组合起来，复杂度可以拆分成 预处理完全平方数 + 组合完全平方数。

0-9 能组合出的数字最大为 9876543210，不超过 1e10，可以枚举 1~1e5 的平方并判断是否满足每个数字最多出现一次，为了方便之后的步骤，我们将满足要求的平方数（无重复数字）的状态储存起来。

• 什么是状态？

数字的状态为一串二进制数字，从右到左第 i 位用于标记 i 这个数字是否出现过，例如 0101001101 的第 0，2，3，6，8 位是 1，代表出现过 0，2，3，6，8 在原始数值中出现过

以下是一些相关的二进制操作：

// x 是待判定的平方数
res = x % 10; // 取出 x 最后一位数字
zt = 1 << res;// 1 左移 res 位，与即将标记 res 的状态位对齐
// status 记录了之前的标记状态
if((status & zt) != 0) ......// 如果 res 出现过，status 和 zt 的二进制的 res 位上都是 1，结果不为 0，那么 x 就不符合条件
status |= zt; // 改变 status 的 res 对应的状态位，标记为 1

 

在得到了所有满足条件的平方数状态之后，可以对这些平方数进行组合，如果一种组合中数字中 0~9 分别出现一次，就满足了题目条件记录答案。题目不考虑不同组合中顺序的问题，所以我们规定按照一定的顺序从前往后组合数字，保证一种组合只出现一次。

假设当前组合的数字状态为 x ，那么枚举可以加入的数字状态 y，因为每一个数字最终只能出现一次，所以任意一位上，x 和 y 不能同时为 1 也就是必须 x & y == 0，否则 y 就不能与 x 组合到一起。

如果 x & y == 0 说明数字 y 可以加入组合 x 中，那么加入后的组合状态应变为 x = x | y;。

后当一种组合包含了 0~9，也就是 1111111111 = (1 << 10) - 1，说明当前组合可行，记录答案。

代码和思路均参考蓝桥杯官方决赛特训营

#include<bits/stdc++.h>

using namespace std;
const int maxn = 1 << 10;
const int Final_status = (1<<10) -1;

vector<int> nums; //存放符合条件平方数，数字的状态
int ans;

void check(long long x) {
    int status = 0;//记录数字的状态，初始时全部标记为0
    if (x == 0)status = 1;//标记最低位即第0位为1
    while (x) {
        //获取10进制个位上的数，1移位后与上一次的状态做与运算
        //不为0说明个位上的数字已经被使用，则不记录，return
        if ((status & (1 << (x % 10))) != 0)return;

        //记录数字使用情况
        status |= (1 << (x % 10));  //标记个位对应位置为1
        x /= 10; //消掉个位
    } //继续处理新的个位，直到全部消掉
    //此时，原始的x的每一位都被标记到了status中且没有重复数字
    nums.push_back(status); //添加到vector中
}

void dfs(int x, int y) {//从第x个合法数字往后取，当前组合状态为y
    if (y == Final_status) {//组合可行，记录答案
        ans++;
        return;
    }
    for (int i = x; i < nums.size(); i++) {//枚举新加入数字的状态
        if ((nums[i] & y) == 0) {//可以加入组合，去找下一个数字
            //dfs(i + 1, nums[i] | y);
            int pre = y;
            y |= nums[i];
            dfs(i+1, y);

            //回溯
            y = pre;
        }
    }
}

int main() {
    //枚举算出满足题意的平方数并储存其状态
    for (long long i = 0; i <= 100000; i++) {
        check(i * i);
    }

    //用深搜按顺序组合预处理出的平方数
    dfs(0, 0);
    cout << ans << endl;
}