Codeforces 797F Mice And Holes DP(单调队列优化)

点击打开链接

题意:n个老鼠,m个洞,n,m<=5000　老鼠在点xi,洞在pj,可以容纳cj只老鼠.i->j花费:|xi-pj|,问n只老鼠都跑到洞里的最小花费? 
定义状态d[i][j]:前i个洞,装前j只老鼠的最小花费 ,sum[i][j]表示前j只老鼠进第i洞话费 
首先排序后,最优的情况为:进一个洞的老鼠一定是连续的一批(不会出现a,c进洞i,b进洞i-1)->(明显a进洞i-1,b,c进洞i更优) 
容易得到:靠右边的洞肯定让前j只中靠右边先进 
前i-1个洞装多少只:d[i][j]=min(d[i-1][k]+sum[i][j]-sum[i][k]) (k=0~j).

O(m*n*n) 优化?:只要快速找到[0~j]中最优的决策
假设x<y 决策y优于决策x dp[i-1][x]+sum[i][j]-sum[i][x]>dp[i-1][y]+sum[i][j]-sum[i][y] 令f[x]=dp[i-1][x]-sum[i][x]
最优决策y满足:f[y]>f[x], j-y<=a[i].c 用单调队列维护即可 O(n*m)

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5002;
const ll inf=1e14;
ll n,m,x[N],sum[N];//sum[i][j] 前j只->hole[i] 
ll dp[N][N];
ll que[N];
struct node{
	ll p,c;
	bool operator <(const node& t)const{
		return p<t.p;
	}
}a[N];
ll f(int i,ll x)
{
	return dp[i-1][x]-sum[x];
} 
int main()
{
	while(cin>>n>>m)
	{
		for(int i=1;i<=n;i++)
			scanf("%I64d",&x[i]);
		sort(x+1,x+n+1);
		for(int i=1;i<=m;i++)
			scanf("%I64d%I64d",&a[i].p,&a[i].c);
		sort(a+1,a+1+m);
		memset(dp,0x3f3f3f3f,sizeof(dp));
		for(int i=0;i<=m;i++)
			dp[i][0]=0;

		
		
		for(ll i=1;i<=m;i++)
		{	
			ll c=a[i].c;
			ll head=1,tail=0;
			que[++tail]=0;
			memset(sum,0,sizeof(sum));
			for(ll j=1;j<=n;j++)
			{
				sum[j]=sum[j-1]+abs(a[i].p-x[j]);
				//队列(下标),递增 
				dp[i][j]=dp[i-1][j];//
				while(j-que[head]>c) head++;
				while(head<=tail&&dp[i-1][que[head]]>=inf) head++;
				
				while(head<tail&&f(i,que[head])>f(i,que[head+1])) head++;//
				
				if(head<=tail)
					dp[i][j]=min(dp[i][j],dp[i-1][que[head]]+sum[j]-sum[que[head]]);
				que[++tail]=j;	
			}
		}
		if(dp[m][n]>=inf)
			puts("-1");
		else
			cout<<dp[m][n]<<endl;
	}
	
	return 0;
}