Description
ar把一段凹凸不平的路分成了高度不同的N段,并用H[i]表示第i段高度。现在Tar一共有n种泥土可用,它们都能覆盖给定的连续的k个部分。
对于第i种泥土,它的价格为C[i],可以使得区间[i,min(n,i+k-1)] 的路段的高度增加E[i]。
Tar要设定一种泥土使用计划,使得使用若干泥土后,这条路最低的高度尽量高,并且这个计划必须满足以下两点要求:
(1)每种泥土只能使用一次。
(2)泥土使用成本必须小于等于M。
请求出这个最低的高度最高是多少。
Input
第一行为如上文所示的三个正整数:N,M,K。
接下来N行,每行3个如上文所示的正整数H[i],E[i],C[i]。
Output
输出有且只有一个数字,为最底部分的高度的最大值
Sample Input
4 20 1
1 3 5
1 7 3
4 6 9
3 5 13
Sample Output
3
Data Constraint
对于30%的数据:N≤20。
对于100%的数据:1≤K≤11,1≤N≤100,0≤M,H[i],E[i],C[i]≤1000000。
思路
二分答案,将问题转化为判定性问题。
注意到1<=K<=11,也就是说对于第i个路段,能够提高它高度的泥土只有从第i-K个到第i个。于是使用状态压缩动态规划求cost。f[i][j]表示前i个路段通过泥土使用>=高度ans,且第i - k个到第i个的使用情况为二进制数j的cost,我的代码是从二进制个位表示i的使用情况,所以得到从f[i-1]到f[i]的转移方程为
f[i][j]=min(f[i-1][j>>1],f[i-1][(j>>1)+pow[k-1]])+(j&1)*c[i];
其中f[i-1][j>>1]表示第i-k-1个泥土没用,f[i-1][(j>>1)+pow[k-1]]表示第i-k-1个泥土用了。
若f[i][j]=INF,则表示该状态前i条路中有没达到预定高度或cost超出了m;
g[j]表示在j状态下i可以升高多少。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=177,inf=0x3f3f3f3f;
int n,m,k,h[maxn],e[maxn],c[maxn],p[20],f[maxn][1<<12],g[1<<12];
int minp(int x)
{
for(int i=k;i>=0;i--) if(x>=p[i]) return i;
}
bool check(int x)
{
memset(f,0x3f,sizeof(f)); f[0][0]=0;
for(int i=1; i<=n; i++)
{
bool b=0;
g[0]=0;
int mx=i<k?p[i]:p[k];
for(int j=0; j<mx; j++)
{
int t;
if(j==0)g[j]=0;
else
{
t=minp(j);
g[j]=g[j-p[t]]+e[i-t];
}
if(g[j]+h[i]<x) continue;
if(f[i-1][j>>1]==inf&&f[i-1][(j>>1)+p[k-1]]==inf) continue;
t=min(f[i-1][j>>1],f[i-1][(j>>1)+p[k-1]]);
f[i][j]=t+(j&1)*c[i];
if(f[i][j]>m) f[i][j]=inf; else b=1;
}
if(!b)return 0;
}
return 1;
}
int main()
{
freopen("cover.in","r",stdin); freopen("cover.out","w",stdout);
scanf("%d%d%d",&n,&m,&k);
p[0]=1;
for(int i=1; i<=k; i++) p[i]=p[i-1]<<1;
for(int i=1; i<=n; i++) scanf("%d%d%d",&h[i],&e[i],&c[i]);
int l=0,r=12000000;
while(l<=r)
{
int mid=(l+r)>>1;
if(check(mid)) l=mid+1;else r=mid-1;
}
printf("%d",r);
}