c++ 魔法阵

六十年一次的魔法战争就要开始了，大魔法师准备从附近的魔法场中汲取魔法能量。

大魔法师有 mm 个魔法物品，编号分别为 1,2,…,m1,2,…,m。

每个物品具有一个魔法值，我们用 xixi 表示编号为 ii 的物品的魔法值。

每个魔法值 xixi 是不超过 nn 的正整数，可能有多个物品的魔法值相同。

大魔法师认为，当且仅当四个编号为 a,b,c,da,b,c,d 的魔法物品满足 xa<xb<xc<xd，xb−xa=2(xd−xc)xa<xb<xc<xd，xb−xa=2(xd−xc)，并且 xb−xa<(xc−xb)/3xb−xa<(xc−xb)/3 时，这四个魔法物品形成了一个魔法阵，他称这四个魔法物品分别为这个魔法阵的 AA 物品，BB 物品，CC 物品，DD 物品。

现在，大魔法师想要知道，对于每个魔法物品，作为某个魔法阵的 AA 物品出现的次数，作为 BB 物品的次数，作为 CC 物品的次数，和作为 DD 物品的次数。

输入格式

输入文件的第一行包含两个空格隔开的正整数 nn 和 mm。

接下来 mm 行，每行一个正整数，第 i+1i+1 行的正整数表示 xixi，即编号为 ii 的物品的魔法值。

保证每个 xixi 是分别在合法范围内等概率随机生成的。

输出格式

共输出 mm 行，每行四个整数。

第 ii 行的四个整数依次表示编号为 ii 的物品作为 A,B,C,DA,B,C,D 物品分别出现的次数。

保证标准输出中的每个数都不会超过 109109。

每行相邻的两个数之间用恰好一个空格隔开。

数据范围

1≤n≤150001≤n≤15000,
1≤m≤400001≤m≤40000,
1≤xi≤n1≤xi≤n

输入样例：

30 8
1
24
7
28
5
29
26
24

输出样例：

4 0 0 0
0 0 1 0
0 2 0 0
0 0 1 1
1 3 0 0
0 0 0 2
0 0 2 2
0 0 1 0

算法
(前缀和,数学,组合计数,乘法原理,加法原理) O(n29)O(n29)
一个魔法阵如下所示：

A和B之间的距离是2t，B和C之间的距离是6t+k，C和D之间的距离是t，其中t, k均为正整数。

左边红色部分框出的A和B是绑定的，右边绿色部分框出的C和D也是绑定的。
因此整个系统共有三个自由度：t、红色部分、绿色部分。

同时枚举三个自由度的计算量过大。在1秒内，我们只能枚举其中两个自由度。

首先枚举t。接下来并列枚举绿色部分和红色部分：

从左到右枚举绿色部分，当绿色部分固定后，则C应该累加的次数是所有满足要求的A和B的 cnt[A] * cnt[B] 的和，再乘以cnt[D]。其中cnt[A], cnt[B], cnt[D]是A, B, D出现的次数。所有满足要求的A和B就是整个线段左边的某个前缀，因此可以利用前缀和算法来加速计算。cnt[D]同理可得。
从右到左枚举红色部分可做类似处理。
时间复杂度
由于A, B, C, D均在1到n之间，因此 1≤t≤n−291≤t≤n−29。所以总枚举次数大约是 n−29∗n=O(n29)=2.5×107n−29∗n=O(n29)=2.5×107。

参考文献
NOIP2016T4 魔法阵。

C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 15010, M = 40010;

int n, m;
int cnt[N];
int a[N], b[N], c[N], d[N], x[M];

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= m; i ++ )
    {
        scanf("%d", &x[i]);
        cnt[x[i]] ++ ;
    }

    for (int t = 1; t * 9 + 2 <= n; t ++ )
    {
        int sum = 0;
        for (int D = t * 9 + 2; D <= n; D ++ )
        {
            int A = D - t * 9 - 1;
            int B = A + t * 2;
            int C = D - t;
            sum += cnt[A] * cnt[B];
            c[C] += sum * cnt[D];
            d[D] += sum * cnt[C];
        }
        sum = 0;
        for (int A = n - t * 9 - 1; A; A -- )
        {
            int B = A + t * 2;
            int D = A + t * 9 + 1;
            int C = D - t;
            sum += cnt[C] * cnt[D];
            a[A] += sum * cnt[B];
            b[B] += sum * cnt[A];
        }
    }

    for (int i = 1; i <= m; i ++ ) printf("%d %d %d %d\n", a[x[i]], b[x[i]], c[x[i]], d[x[i]]);
    return 0;
}