【ACWing】98. 分形之城

题目地址：

https://www.acwing.com/problem/content/100/

城市的规划在城市建设中是个大问题。不幸的是，很多城市在开始建设的时候并没有很好的规划，城市规模扩大之后规划不合理的问题就开始显现。而这座名为Fractal的城市设想了这样的一个规划方案，如下图所示：

当城区规模扩大之后，Fractal的解决方案是把和原来城区结构一样的区域按照图中的方式建设在城市周围，提升城市的等级。对于任意等级的城市，我们把正方形街区从左上角开始按照道路标号。虽然这个方案很烂，Fractal规划部门的人员还是想知道，如果城市发展到了等级$N$，编号为$A$和$B$的两个街区的直线距离是多少。街区的距离指的是街区的中心点之间的距离，每个街区都是边长为$10$米的正方形。

输入格式：
第一行输入正整数$n$，表示测试数据的数目。以下$n$行，输入$n$组测试数据，每组一行。每组数据包括三个整数$N,A,B$，表示城市等级以及两个街区的编号，整数之间用空格隔开。

输出格式：
一共输出$n$行数据，每行对应一组测试数据的输出结果，结果四舍五入到整数。

数据范围：
$1\le N\le 31$
$1\le A,B\le 2^{2N}$
$1\le n\le 1000$

以最左上角的房子为坐标原点，竖直向下是$x$轴，水平向右是$y$轴，建立坐标系（按照二维数组的习惯建立坐标系）。考虑如何递归求解。设$f(n,s)$是$s$号房屋在第$n$个等级里的坐标。首先当$N=0$的时候，房子的坐标是$(0,0)$，所以$f(0,0)=(0,0)$（为了方便我们从$0$开始对房屋计数）。当$N=n$的时候，一共有$2^{2n}$个房屋，分成左上、右上、左下和右下一共$4$个部分，每个部分有$2^{2n-2}$个房屋。我们可以先将$s$对$2^{2n-2}$取余，这样就得到了它在第$n-1$等级中的坐标，再求一下$\lfloor \frac{s}{2^{2n-2}}\rfloor$，这就求出了它在左上、右上、左下还是右下，$0$对应左上，$1$对应右上，$2$对应左下，$3$对应右下。如果在右上，则只需要将$f(n-1,s\mathrm{m}\mathrm{o}\mathrm{d}{2}^{2n-2})$向右平移$2^{n-1}$即可，即$y$增加$2^{n-1}$；如果在右下，则$x$和$y$同时增加$2^{n-1}$；如果在左上，则$x$和$y$交换位置；如果在左下，则做变换$(x,y)\to (2^n-1-y,2^{n-1}-1-x)$，其实就是求$(x,y)$关于$x=-y+2^n-1$的对称点，设这个对称点是$(x^{\prime },y^{\prime })$，那么应该有$x+x^{\prime }=-(y+y^{\prime })+2^{n+1}-2$，还有斜率关系$x-x^{\prime }=y-y^{\prime }$解之即可。这样就可以递归求解了。代码如下：

#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;

struct Point {
    long x, y;
};

Point get(long n, long a) {
    if (n == 0) return {0, 0};
    long block = 1l << 2 * n - 2, len = 1l << n - 1;
    auto p = get(n - 1, a % block);
    long x = p.x, y = p.y;

    int z = a / block;
    if (!z) return {y, x};
    else if (z == 1) return {x, y + len};
    else if (z == 2) return {x + len, y + len};
    else return {len * 2 - 1 - y, len - 1 - x};
}

int main() {
    int T;
    scanf("%d", &T);
    while (T--) {
        long n, a, b;
        scanf("%ld%ld%ld", &n, &a, &b);
        auto pa = get(n, a - 1), pb = get(n, b - 1);
        double dx = pa.x - pb.x, dy = pa.y - pb.y;
        printf("%.0lf\n", sqrt(dx * dx + dy * dy) * 10);
    }

    return 0;
}

时空复杂度$O(N)$。