题意: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;
}