想要强化自己的DP能力,于是找了几篇单调队列的论文,开始看论文做题目,但是结果第一题就悲剧了,做了好久,今天终于A掉了。
状态转移方程
f[i,j]=min(f[x][p]-sum[i][p]+k)+sum[i][j] (x!=i) f[i,j]表示完成前j个流程期中最后一个流程在i上完成的最小耗费
sum[i][j] 机器i完成前j个流程的总时间
k从一台机器转移到另一条机器上的时间。
刚开始纠结这道题目怎么用单调队列,论文里面也只是给出了递推的方程,后来发现要用五个单调队列。分别保存可以转移到与单调队列编号相同的机器上的状态。因为一台机器最多不能同时工作l个流程,这与之前一道单调队列滑动窗口求最值的题目异曲同工,
另外,如果你确定状态转移的地方没有错误的话,那么看看单调队列有没有写错,我在这里浪费了很多时间。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define INF 0xFFFFFFF
#define MAXN 200100
long long a[6][MAXN];
int dque[6][MAXN],t[6],h[6]; //h=0,t=-1;
long long b[6][MAXN];
int m,n,k,l;
long long dp[6][MAXN];
void insertMin(int n,int i)
{
while(h[n]<=t[n] && a[n][dque[n][t[n]]]>=a[n][i])
t[n]--;
dque[n][++t[n]]=i;
}
long long gethead(int n,int i)
{
while(h[n]<=t[n] && dque[n][h[n]]<i-l)
h[n]++;
return a[n][dque[n][h[n]]];
}
void solve()
{
for(int i=1;i<=n;i++)
{
b[i][0]=0;
for(int j=1;j<=m;j++)
{
scanf("%lld",&b[i][j]);
// cin>>b[i][j];
b[i][j]+=b[i][j-1];
}
}
if(n==1)
printf("%lld\n",b[1][m]);
memset(h,0,sizeof(h)),memset(t,-1,sizeof(t));
for(int i=1;i<=n;i++)
{
dp[i][0]=0;
a[i][0]=0;
insertMin(i,0);
}
for(int i=1;i<=m;i++)
{
for(int j=1;j<=n;j++)
dp[j][i]=gethead(j,i)+b[j][i];
for(int j=1;j<=n;j++)
{
a[j][i]=INF;
for(int l=1;l<=n;l++)
{
if(l==j) continue;
a[j][i]=min(dp[l][i],a[j][i]);
}
a[j][i]=a[j][i]-b[j][i]+k;
insertMin(j,i);
}
}
long long ans=dp[1][m];
for(int i=1;i<=n;i++)
ans=min(ans,dp[i][m]);
printf("%lld\n",ans);
}
int main()
{
while(~scanf("%d%d%d%d",&m,&n,&k,&l))
solve();
return 0;
}