标签:区间DP
题目
Description
有n家洗车店从左往右排成一排,每家店都有一个正整数价格p[i]。
有m个人要来消费,第i个人会驶过第a[i]个开始一直到第b[i]个洗车店,且会选择这些店中最便宜的一个进行一次消费。
但是如果这个最便宜的价格大于c[i],那么这个人就不洗车了。
请给每家店指定一个价格,使得所有人花的钱的总和最大。
Input
第一行包含两个正整数\(n,m(1\leq n\leq 50,1\leq m\leq 4000)\)。
接下来m行,每行包含三个正整数\(a[i],b[i],c[i](1\leq a[i]\leq b[i]\leq n,1\leq c[i]\leq 500000)\)
Output
第一行输出一个正整数,即消费总额的最大值。
第二行输出n个正整数,依次表示每家洗车店的价格\(p[i]\),要求\(1\leq p[i]\leq 500000\)。
若有多组最优解,输出任意一组。
Sample Input
7 5
1 4 7
3 7 13
5 6 20
6 7 1
1 2 5
Sample Output
43
5 5 13 13 20 20 13
分析
首先将\(c\)离散化,显然\(p[i]\)一定存在于\(c\)中
设状态\(g[i][j][k]\)表示区间\(i->j\)中所有花费\(\geq k\)的最大收益
\(h[x][k]\)表示经过x点的\(c_i\geq k\)的人数
\(f[i][j][k]\)为区间[i,j]最小值为k的最优方案(仅仅记录下来)
和区间DP转移类似,枚举分割点x
\[g[i][j][k]=g[i][l-1][k]+g[l+1][j][k]+v[k]*h[x][k]\]记录方案后通过dfs输出
code
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read(){
ll f=1,x=0;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
const int maxn=56,maxm=4e3+6;
int n,m,a[maxm],b[maxm],c[maxm],v[maxm],h[maxn][maxm];
char f[maxn][maxn][maxm];int g[maxn][maxn][maxm],p[maxn][maxn][maxm];
void dfs(int l,int r,int k){
if(l>r)return;
int x=f[l][r][k=p[l][r][k]];
a[x]=v[k],dfs(l,x-1,k),dfs(x+1,r,k);
}
inline int lower(int x){
int l=1,r=m,mid,tmp;
while(l<=r)if(v[mid=(l+r)>>1]<=x)l=(tmp=mid)+1;else r=mid-1;
return tmp;
}
int main(){
n=read(),m=read();
rep(i,1,m)a[i]=read(),b[i]=read(),v[i]=c[i]=read();
sort(v+1,v+1+m);
rep(i,1,m)c[i]=lower(c[i]);
dep(i,n,1)
rep(j,i,n){
rep(k,i,j)rep(x,1,m)h[k][x]=0;
mem(h,0);
rep(k,1,m)if(i<=a[k]&&b[k]<=j)rep(x,a[k],b[k])h[x][c[k]]++;
rep(k,i,j)dep(x,m-1,1)h[k][x]+=h[k][x+1];
dep(k,m,1){
int x,y;
for(x=i,y=0;x<=j;x++){
int tmp=g[i][x-1][k]+g[x+1][j][k]+v[k]*h[x][k];
if(tmp>=y)y=tmp,f[i][j][k]=x;
}
if(y>=g[i][j][k+1])g[i][j][k]=y,p[i][j][k]=k;
else g[i][j][k]=g[i][j][k+1],p[i][j][k]=p[i][j][k+1];
}
}
dfs(1,n,1);
printf("%d\n",g[1][n][1]);
rep(i,1,n)printf("%d ",a[i]);puts("");
return 0;
}