NOIP2016P4 魔法阵 题解

（题目描述略）

设魔法值为 j 物品 $nu{m}_j$ 个，则魔法阵 $x_a,x_b,x_c,x_d$ 对于 A, B, C, D 物品贡献分别为 $nu{m}_{xb}\ast nu{m}_{xc}\ast nu{m}_{xd},nu{m}_{xa}\ast nu{m}_{xc}\ast nu{m}_{xd},nu{m}_{xa}\ast nu{m}_{xb}\ast nu{m}_{xd},nu{m}_{xa}\ast nu{m}_{xb}\ast nu{m}_{xc}$。设 $i=x_d-x_c$，则 $x_b-x_a=2i$，$x_c-x_b>6i$，$x_d-x_b>7i$，$x_c-x_a>8i$，$x_d-x_a>9i$。枚举 $i\in [1,\lfloor \frac{n-1}{9}\rfloor ]$，并枚举 $x_a,x_d$ 使得 $1\le x_a\le x_d\le n$ 且 $x_d-x_a>9i$，由 $x_b=x_a+2i$ 和 $x_c=x_d-i$ 得出每组可能 $x_a,x_b,x_c,x_d$ 统计即可，这样可得 85 分。

对于满分算法，考虑当 $x_a=n-9i-1$，即 $x_b=n-7i-1$ 时，相应 $\{\begin{array}{l}{x}_c=n-i\\ x_d=n\end{array}$；当 $\{\begin{array}{l}{x}_a=n-9i-2\\ x_b=n-7i-2\end{array}$ 时，相应 $\{\begin{array}{l}{x}_c=n-i\\ x_d=n\end{array}$ 或 $\{\begin{array}{l}{x}_c=n-1-i\\ x_d=n-1\end{array}$。一般的，当 $\{\begin{array}{l}{x}_a=n-9i-j\\ x_b=n-7i-j\end{array}(1\le j<n-9i)$ 时，相应 $x_c,x_d$ 解集为 $\{\begin{array}{l}{x}_a=n-9i-j+1\\ x_b=n-7i-j+1\end{array}$ 解集和 $\{\begin{array}{l}{x}_c=n-j+1-i\\ x_d=n-j+1\end{array}$ 的并集，对于 A, B 物品贡献分别为 $nu{m}_{xb}\ast {\sum }_{k=1}^j(nu{m}_{n-k+1-i}\ast nu{m}_{n-k+1})$ 和 $nu{m}_{xa}\ast {\sum }_{k=1}^j(nu{m}_{n-k+1-i}\ast nu{m}_{n-k+1})$，而 ${\sum }_{k=1}^j(nu{m}_{n-k+1-i}\ast nu{m}_{n-k+1})={\sum }_{k=1}^{j-1}(nu{m}_{n-k+1-i}\ast nu{m}_{n-k+1})+nu{m}_{n-j+1-i}\ast nu{m}_{n-j+1}$，前者可递推得到。

代码如下：

#include<stdio.h>
#define MAX_M (40005)
#define MAX_N (15005)
int num[MAX_N],suma[MAX_N],sumb[MAX_N],sumc[MAX_N],sumd[MAX_N],x[MAX_M];
int main()
{
	freopen("magic.in","r",stdin);
	freopen("magic.out","w",stdout);
	int k,m,n;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		num[i]=suma[i]=sumb[i]=sumc[i]=sumd[i]=0;
	for(int i=1;i<=m;i++)
		scanf("%d",&x[i]),num[x[i]]++;
	for(int i=1;i*9<n-1;i++)
	{
		k=0;
		for(int id=n;id-i*9>1;id--)
			k+=num[id-i]*num[id],
			suma[id-i*9-1]+=num[id-i*7-1]*k,
			sumb[id-i*7-1]+=num[id-i*9-1]*k;
		k=0;
		for(int ia=1;ia+i*9<n;ia++)
			k+=num[ia]*num[ia+i*2],
			sumc[ia+i*8+1]+=num[ia+i*9+1]*k,
			sumd[ia+i*9+1]+=num[ia+i*8+1]*k;
	}
	for(int i=1;i<=m;i++)
		printf("%d %d %d %d\n",suma[x[i]],sumb[x[i]],sumc[x[i]],sumd[x[i]]);
	return 0;
}