【bzoj4380】[POI2015]Myjnie

题目

将ci离散，显然pi一定是某个cj。
设g[i][j][k]代表i，j区间所有花费>=k的最优解。
枚举分割点x，然后g[i][j][k]=g[i][l-1][k]+g[l+1][j][k]+v[k]*h[x][k]
h[x][k]代表经过x点的ci>=k的人数。
然后区间dp转移，记录方案，dfs输出。

代码：

#include
#include
#include
 
#define maxn 52
#define maxm 4010
 
using namespace std;
 
int n,m,t[maxm],f[maxn][maxn][maxm],g[maxn][maxn][maxm],h[maxn][maxm],M[500010],num,inv[maxm],pre[maxn][maxn][maxm];
int l[maxm],r[maxm],w[maxm],ans[maxn];
 
void dfs(int l,int r,int k)
{
    if(l>r) return;
    int mid=f[l][r][k],t=pre[l][r][k];
    ans[mid]=inv[t];
    dfs(l,mid-1,t);
    dfs(mid+1,r,t);
}
 
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&l[i],&r[i],&w[i]);
        t[++t[0]]=w[i];
    }
    sort(t+1,t+m+1);
    for(int i=1;i<=m;i++) if(t[i]!=t[i-1]||i==1) M[t[i]]=++num,inv[num]=t[i];
    for(int i=1;i<=m;i++) w[i]=M[w[i]];
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            for(int k=1;k<=m;k++) g[i][j][k]=-0x3fffffff;
        }
    }
    for(int i=n;i>0;i--){
        for(int j=i;j<=n;j++){
            for(int k=1;k<=m;k++){
                for(int x=i;x<=j;x++){
                    h[x][k]=0;
                }
            }
            for(int k=1;k<=m;k++){
                if(l[k]>=i&&r[k]<=j){
                    for(int x=l[k];x<=r[k];x++){
                        h[x][w[k]]++;
                    }
                }
            }
            for(int x=i;x<=j;x++){
                for(int k=m-1;k>0;k--){
                    h[x][k]+=h[x][k+1];
                }
            }
            for(int x=i;x<=j;x++){
                for(int k=1;k<=m;k++){
                    if(g[i][x-1][k]+g[x+1][j][k]+inv[k]*h[x][k]>g[i][j][k]){
                        g[i][j][k]=g[i][x-1][k]+g[x+1][j][k]+inv[k]*h[x][k];
                        f[i][j][k]=x,pre[i][j][k]=k;
                    }
                }
            }
            for(int k=m;k>0;k--){
                if(g[i][j][k+1]>g[i][j][k]){
                    g[i][j][k]=g[i][j][k+1];
                    pre[i][j][k]=pre[i][j][k+1];
                    f[i][j][k]=f[i][j][k+1];
                }
            }
        }
    }
    dfs(1,n,1);
    printf("%d\n",g[1][n][1]);
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}