dp[l]表示得到长度为l的rubber花费的最小cost,那么对于rubber i,[Ai,Bi],如果已经有一个rubber 长度在[l-Bi,l-Ai]范围内,那么现有的rubber加上rubber i就可以得到长度为l的rubber。对应的状态转移方程即为dp[l]=min(dp[l-Bi],...,dp[l-Ai])+cost of rubber i
因为rubber i的cost是固定的,如果要找到dp[l-Bi],...,dp[l-Ai]的最小值,可以通过维护一个sliding window实现。对于每一个rubber i,窗口大小即为Bi-Ai+1。之后在针对每一个l,从右往左滑动窗口,窗口可以用一个双端队列来维护,left side of queue is front, right side is back。queue是个单调队列,最左端的元素值最大。
初始化时,把[L-Bi,L-Ai]的最小元素push_front到queue中,窗口向左移动一格(l--),新加入的元素即为l-Bi,如果l-Bi比queue最左端元素小,就将queue pop_front直到l-Bi比queue最左端元素大or相等为止。再讲l-Bi push_front到queue中。
如果queue右端元素不在窗口范围内,就pop_back。这样,对于每一个l,queue最右端元素即为窗口中的最小元素。
这一题WA了好久,是因为没有初始化==。这样对于窗口[L-Bi,L-Ai]push到queue中的就不是最小值。>_<上述滑动窗口push/pop原理都是基于前一个窗口已经存在做的。
#include<iostream>
#include<stdio.h>
#include<cstdio>
#include<string>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
using namespace std;
//2017 RoundD Problem D. Stretch Rope
const int maxn=1010;
int T;
int N;
int M;
int L;
int A[maxn];
int B[maxn];
int P[maxn];
int dp[10010];
int main()
{
freopen("D-large-practice.in","r",stdin);//input.txt
freopen("output.txt","w",stdout);
scanf("%d",&T);
for(int ca=1;ca<=T;ca++)
{
scanf("%d %d %d",&N,&M,&L);
memset(A,0,sizeof(A));
memset(B,0,sizeof(B));
memset(P,0,sizeof(P));
memset(dp,0x3f,sizeof(dp));
for(int i=0;i<N;i++)
{
scanf("%d %d %d",&A[i],&B[i],&P[i]);
}
dp[0]=0;//get a rubber of length 0 needs no money
for(int i=0;i<N;i++)
{
//cout<<i<<endl;
deque<int>que;//save the index of min element in the window
while(!que.empty()) que.pop_back();
int winsz=B[i]-A[i]+1;
for(int j=L-A[i];j>=max(0,L-B[i]);j--)//initialize, push the elements of first window into queue
{
while(que.size()>0&&dp[que.front()]>dp[j])
{
que.pop_front();
}
que.push_front(j);
}
//for(int l=A[i];l<=L;l++) dp[l] is transferred from dp[m], m<l, this may cause select rubber i twice
for(int l=L;l>=A[i];l--)//the window moves towards left
{
while(que.size()>0&&l>=B[i]&&dp[que.front()]>dp[l-B[i]])
{
que.pop_front();
}
while(que.size()>0&&que.back()-l+B[i]+1>winsz)
{
que.pop_back();
}
if(l>=B[i])
{
que.push_front(l-B[i]);
}
int idx=que.back();
// if(l-A[i]>=0)
// {
dp[l]=min(dp[l],dp[idx]+P[i]);
//cout<<l<<" "<<idx<<" "<<dp[l]<<" "<<dp[idx]<<endl;
// }
// int Min = 0x3f3f3f3f;
// for(int k = l - A[i];k >= l - B[i] && k >=0;k--)
// Min = min(dp[k],Min);
// //if(Min != 0x3f3f3f3f)
// dp[l] = min(dp[l],Min + P[i]);
}
}
if(dp[L]>M)
{
printf("Case #%d: IMPOSSIBLE\n",ca);
}
else
{
printf("Case #%d: %d\n",ca,dp[L]);
}
}
return 0;
}