[NOI2002] 贪吃的九头蛇

考虑任意一种划给大头的方案,两端的都给了大头(bel=1)的边产生难受值,剩下n-k个果子分给m-1个头,当m-1=1时,两端都给了这个小头也产生难受值;而m-1>1的情况要好看的多,贪心的,因为未划分的果子构成一个森林,重新计算这些果子在所在树中的深度,把果子按深度排序,前m-1个个分别划分,剩下的节点任意分配,只需要保证与父亲不同即可,显然按这种分配方法能做到“零难受”。

换而言之,当m=2时只有两端分配不同才不会有难受值(废话!);否则,只有两端都分配给了1(大头)才会有难受值。然后就能写dp了,设f[x,0/1]表示x子树中,x是否划分给了1的最小难受值之和。

\[f[x,0]=\begin{cases} +\infty &x\text{是最大值点}\\ \sum_{x\to y} \min(f[y,0]+len_{x\to y},f[y,1]) &m=2\\ \sum_{x\to y} \min(f[y,0],f[y,1]) &m>1 \end{cases}\\ f[x,1]=\sum_{x\to y} \min(f[y,0],f[y,1]+len_{x\to y}) \]

似乎很可信的样子……可行个喘喘,怎么着也得加一位表示子树内已经分配给1的节点总数吧……反正不再列式子了(光速逃

#include <bits/stdc++.h>
using namespace std;
const int N=310;

int n,m,k;
int head[N],to[N<<1],len[N<<1],lst[N<<1];
int siz[N],f[N][N][2],tmp[N][2];

void ins(int x,int y,int w) {
    static int cnt=0;
    to[++cnt]=y,len[cnt]=w,lst[cnt]=head[x],head[x]=cnt;
    to[++cnt]=x,len[cnt]=w,lst[cnt]=head[y],head[y]=cnt;
}
void dfs(int x,int pa) {
    siz[x]=1;
    memset(f[x],0x3f,sizeof f[x]);
    f[x][0][0]=f[x][1][1]=0;
    for(int i=head[x]; i; i=lst[i]) if(to[i]!=pa) {
        int y=to[i];
        dfs(y,x); siz[x]+=siz[y];
        memcpy(tmp,f[x],sizeof f[x]);
        memset(f[x],0x3f,sizeof f[x]);
        for(int s=0; s<=k&&s<=siz[x]; ++s) 
        for(int t=0; t<=s&&t<=siz[y]; ++t) {
            f[x][s][0]=min(f[x][s][0],min(f[y][t][0]+(m==2)*len[i],f[y][t][1])+tmp[s-t][0]);
            f[x][s][1]=min(f[x][s][1],min(f[y][t][0],f[y][t][1]+len[i])+tmp[s-t][1]);
        }
    }
}

int main() {
    scanf("%d%d%d",&n,&m,&k);
    for(int x,y,w,i=n; --i; ) {
        scanf("%d%d%d",&x,&y,&w);
        ins(x,y,w);
    }
    if(m-1+k>n) {
        puts("-1");
        return 0;
    }
    dfs(1,0);
    printf("%d\n",f[1][k][1]);
    return 0;
}
posted @ 2019-05-23 17:04  nosta  阅读(226)  评论(0编辑  收藏  举报