NOIP2016 普及组第四题 魔法阵magic 题解

题目描述

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

大魔法师有m个魔法物品，编号分别为1,2,…,m。每个物品具有一个魔法值，我们用Xi表示编号为i的物品的魔法值。每个魔法值Xi是不超过n的正整数，可能有多个物品的魔法值相同。

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

现在，大魔法师想要知道，对于每个魔法物品，作为某个魔法阵的A物品出现的次数，作为B物品的次数，作为C物品的次数，和作为D物品的次数。

输入输出格式

输入格式：
输入文件的第一行包含两个空格隔开的正整数n和m。

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

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

输出格式：
共输出m行，每行四个整数。第i行的四个整数依次表示编号为i的物品作 为A,B,C,D物品分别出现的次数。

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

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

输入输出样例

输入样例#1：
30 8
1
24
7
28
5
29
26
24
输出样例#1：
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
输入样例#2：
15 15
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
输出样例#2：
5 0 0 0
4 0 0 0
3 5 0 0
2 4 0 0
1 3 0 0
0 2 0 0
0 1 0 0
0 0 0 0
0 0 0 0
0 0 1 0
0 0 2 1
0 0 3 2
0 0 4 3
0 0 5 4
0 0 0 5

题解

对于一道普及组的题目来说，这道题的难度还是有点大的
首先要读懂题，题目中有大写X和小写x，（在上面我已经改了），其实是一样的，可能会有人因此混淆

列出条件
1.$xa<xb<xc<xd$
2.$xb-xa=2(xd-xc)$
3.$xb-xa<(xc-xb)/3$

---

先考虑稍微暴力点的方法
对于第一个条件，直接将所有东西存进一个桶里，桶的大小不会超过N（15000），每次不是枚举第几个魔法值，而是枚举魔法值
第二个条件的意思是后前两个数的差等于后两个数的差的两倍
可以利用条件三加一个小优化，这样在统计答案的时候也就不用判断是否满足条件三了
设后两个数的差为