UVALive 3983(单调队列优化dp)

链接：点击打开链接

题意：有n个垃圾，第i个垃圾坐标为(xi,yi)，重量为wi，有一个机器人，要按照编号从小到大的顺序剑气所有的垃圾兵扔进垃圾桶，垃圾桶在源点，每次总重量不能超过C,两点间距离为曼哈顿距离，求出最短的距离和

代码：

#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int q[100005],dp[100005];
int x[100005],y[100005],w[100005];
int dis[100005],sum_dis[100005],sum_w[100005];
int get(int x){
    return dp[x]-sum_dis[x+1]+dis[x+1];
}
int main(){
    int t,n,m,i,j,head,tail;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&m,&n);
        x[0]=y[0]=0;
        sum_w[0]=sum_dis[0]=0;
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++){
            scanf("%d%d%d",&x[i],&y[i],&w[i]);
            dis[i]=abs(x[i])+abs(y[i]);
            sum_w[i]=sum_w[i-1]+w[i];
            sum_dis[i]=sum_dis[i-1]+abs(x[i]-x[i-1])+abs(y[i]-y[i-1]);
        }                                       //dp[i]表示到第i个的花费
        head=0,tail=1;                          //dp[i]=min(dp[j]+sum[i]-sum[j+1]+dis[i]+dis[j])
        q[0]=0;                                 //所以可以用单调队列维护dp[j]-sum_dis[j+1]+dis[j+1]
        for(i=1;i<=n;i++){
            while(head<tail&&sum_w[i]-sum_w[q[head]]>m)
            head++;
            dp[i]=get(q[head])+sum_dis[i]+dis[i];
            while(head<tail&&get(i)<=get(q[tail-1]))
            tail--;
            q[tail++]=i;
        }
        printf("%d\n",dp[n]);
        if(t>0)
        printf("\n");
    }
    return 0;
}