UVALive 3983(单调队列优化dp)

链接:点击打开链接

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

代码:

#include 
#include 
#include 
#include 
#include 
#include 
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(headm)
            head++;
            dp[i]=get(q[head])+sum_dis[i]+dis[i];
            while(head0)
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(动态规划,---------dp优化)