JZOJ 3920.【NOIP2014模拟11.3】噪音

$Description$

给定$n$头牛和$m$个房间，第$i$头牛会进入$p_i$这个房间，在$x$这头牛进入$y$这个房间时，$y$里所有的牛以及这头牛自己都会为$x$这头牛欢呼

这些欢呼会消耗代价，你可以选择在某头牛欢呼完之后清空这个牛棚，求最小欢呼总代价

数据范围：$n\le 10^6,m\le 10^2,k\le 5\times 10^2$

---

$Solution$

首先注意到这些牛棚是互不相关的

于是考虑$dp$

设$f[i][j]$表示处理到第$i$个牛棚，恰好用了$j$次清除的最小代价

问题就转化称为对于每个牛棚，每个选择清除$a_p$次产生$b_p$的代价，要使$\sum a=k$的前提下$b_p$最小（为什么是$\sum a=k$呢？因为删除肯定越多越好，恰好为$k$次必然更优）

于是$dp$

f [ i ] [ j ] = m i n { f [ i − 1 ] [ k ] + w e i g h t ( e d [ i ] , j − k + 1 ) } f[i][j]=min\{f[i-1][k]+weight(ed[i],j-k+1)\} f[i][j]=min{f[i−1][k]+weight(ed[i],j−k+1)}

$weight(n,q)$表示将一个有$n$头牛的牛棚分为$q$份（即删除$q-1$次）的代价，这个可以$O(1)$算，具体解释请移步$wyc$大爷的博客

于是我们就$O(m^{2}k)$解决了这道题

---

$Code$

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm> 
#define LL long long
using namespace std;;
LL f[101][501],n,m,k,x,ed[101];
inline LL read()
{
	char c;LL d=1,f=0;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
inline LL weight(LL gs,LL cs)
{
	LL ls=gs/cs,ms=gs%cs;
	return (1+ls)*ls/2*cs+ms*(ls+1);
}
signed main()
{
	memset(f,0x3f,sizeof(f));
	n=read();m=read();k=read();
	for(register int i=1;i<=n;i++) x=read(),ed[x]++;
	f[0][0]=0;
	for(register int i=1;i<=m;i++)
	 for(register int j=0;j<=k;j++)
	  for(register int p=0;p<=j;p++)
	   f[i][j]=min(f[i][j],f[i-1][p]+weight(ed[i],j-p+1));
	printf("%lld",f[m][k]);
}