题解 【NOIP2016】魔法阵
【NOIP2016】魔法阵
Description
六十年一次的魔法战争就要开始了，大魔法师准备从附近的魔法场中汲取魔法量。
大魔法师有m个魔法物品，编号分别为1,2,...,m。每个物品具有一个魔法值，我们用xi表示编号为i的物品的魔法值。每个魔法值xi是不超过n的正整数，可能有多个物品的魔法值相同。
他称这四个魔法物品分别为这个魔法阵的A物品，B物品，C物品，D物品。
现在，大魔法师想要知道，对于每个魔法物品，作为某个魔法阵的A物品出现的次数，作为B物品的次数，作为C物品的次数，和作为D物品的次数。
Input
第一行包含两个空格隔开的正整数n和m
接下来m行，每行一个正整数，第i+1行的正整数表示x}，即编号为i的物品的魔法值。
保证1 <= n <= 15000,1 <= m <= 40000,1 <= xi <= n。每个xi是分别在合法范围内等概率随机生成的。
Output
共输出m行，每行四个整数。第i行的四个整数依次表示编号为i的物品作为A,B,C,D物品分别出现的次数。
保证标准输出中的每个数都不会超过10^9
每行相邻的两个数之间用恰好一个空格隔开。
Sample Input
输入样例1：
30 8
1
24
7
28
5
29
26
24
输入样例2：
15 15
1
2
3
4
5
6
7
8
9
1
11
12
13
14
15
Sample Output
输出样例1：
4 0 0 0
0 0 1 0
0 2 0 0
0 0 1 1
1 3 0 0
0 0 2 0
0 0 2 2
0 0 1 0
输出样例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
Hint
输入样例1提示：
共有5个魔法阵，分别为:
物品1,3,7,6，其魔法值分别为1, 7, 26, 29;
物品1,5,2,7，其魔法值分别为1, 5, 24, 26;
物品1,5,7,4，其魔法值分别为1, 5, 26, 28;
物品1,5,8,7，其魔法值分别为1, 5, 24, 26;
物品5,3,4,6，其魔法值分别为5, 7, 28, 29 0
以物品_5为例，它作为A物品出现了1次，作为B物品出现了3次，没有作为C物品或者D物品出现，所以这一行输出的四个数依次为1,3,0,0 0
此外，如果我们将输出看作一个m行4列的矩阵，那么每一列上的m个数之和都应等于魔法阵的总数。所以，如果你的输出不满足这个性质，那么这个输出一定不正确。你可以通过这个性质在一定程度上检查你的输出的正确性。
数据范围：
Source
NOIP2016普及组
动态规划
解析
这题用暴力特别好写。
但时间复杂度会炸啊啊啊！！！
所以我们要考虑优化。
首先，想想n是干什么的？
没错，我们可以用桶排！
用d[i]记录i魔法值为i的物品个数。
然后，再考虑一下解法：
先看下图（图片来自网络（真的画不出来））：
当d-c等于t时，各点之间的距离就如上图。
所以，确定d的位置后，c的位置也就确定了。
而对于一对确定的c,d，
能够于它们组成魔法阵的a,b的组数为∑（d[ai]*d[bi]）(ai,bi为满足条件的魔法值)。
因此，我们还能用前缀和优化。
因为当t一定时，若一对（a,b）满足条件，则a,b之前的所有差值为2×t的魔法值都一定满足条件。
最后，c的组数就等于（a,b的组数）×魔法值为d的物品个数（d也一样）。
于是我们枚举d的值，再一边递推就行了。
最后，同理，枚举a的位置，计算满足条件的c,d组数，就可以AC了！（注意，a要倒着枚举，因为c,d是从后往前递推的。）
上AC代码（如果前面不太懂可以看代码理解）：
#include <bits/stdc++.h> using namespace std; int n,m; int a[40001]; int f[5][15001]/*作为第i个物品时,魔法值为j的次数*/; int d[15001]/*魔法值为i的物品个数*/; int sum=0,cnt=0; int main(){ memset(f,0,sizeof(f)); scanf("%d%d",&n,&m); for(int i=1;i<=m;i++){ scanf("%d",&a[i]); d[a[i]]++; } for(int t=1;t*9+2<=n;t++){ sum=0;cnt=0; for(int dd=9*t+2;dd<=n;dd++){ int cc=dd-t;//第三个物品 int bb=cc-6*t-1;//第二个物品 int aa=bb-2*t;//第一个物品 sum+=d[aa]*d[bb];//递推计算前缀和 f[3][cc]+=sum*d[dd];//魔法值为d的个数可能不是/不止1个 f[4][dd]+=sum*d[cc];//同理 } for(int aa=n-9*t-1;aa>=1;aa--){ int bb=aa+2*t;//第二个物品 int cc=bb+6*t+1;//第三个物品 int dd=cc+t;//第四个物品 cnt+=d[cc]*d[dd]; f[1][aa]+=cnt*d[bb]; f[2][bb]+=cnt*d[aa]; } } for(int i=1;i<=m;i++){ for(int j=1;j<=4;j++){ printf("%d ",f[j][a[i]]); } printf("\n"); } return 0; }
