首页 > 代码库 > rivers ioi2005 树形dp
rivers ioi2005 树形dp
说句实话,写完这道题,很想吐一口血出来,以示我心情的糟糕;
题目很简单,树形dp,正常做30分钟,硬是做了好几个小时,真是伤心。
题解不写了,只是吐个槽,网上没有用背包写的dp,全是左儿子右兄弟写法,没法对照一下,难受啊。
代码:
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 using namespace std; 5 #define LL long long 6 int n,K; 7 const int maxn=210; 8 const LL inf=3000000000LL; 9 struct node{ 10 int y,next,v; 11 }e[maxn]; 12 int linkk[maxn],len=0,w[maxn],d[maxn][maxn],siz[maxn],q[maxn],tail=0,fa[maxn],ru[maxn],vis[maxn]; 13 int read(){ 14 int x=0;char ch=getchar();bool flag=0; 15 while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)flag=1;ch=getchar();} 16 while(ch<=‘9‘&&ch>=‘0‘){x=x*10+ch-‘0‘;ch=getchar();} 17 return flag?-x:x; 18 } 19 void insert(int x,int y,int v){ 20 e[++len].y=y; 21 e[len].v=v; 22 e[len].next=linkk[x]; 23 linkk[x]=len; 24 } 25 void print(int x){printf("%d\n",x);} 26 void print(int x,int y){printf("%d %d\n",x,y);} 27 void init(){ 28 int y,v; 29 n=read(),K=read(); 30 memset(d,10,sizeof(d)); 31 for(int i=1;i<=n;i++){ 32 w[i]=read(),y=read(),v=read(); 33 insert(i,y,v);insert(y,i,v); 34 d[i][y]=v,d[y][i]=v; 35 } 36 37 } 38 void findd(){ 39 for(int i=1;i<=n;i++)d[i][i]=0; 40 for(int k=0;k<=n;k++) 41 for(int i=0;i<=n;i++) 42 for(int j=0;j<=n;j++) 43 if(d[i][k]+d[k][j]<d[i][j])d[i][j]=d[i][k]+d[k][j]; 44 } 45 LL f[110][60][110]; 46 void dfs(int x,int f){ 47 siz[x]=1; 48 fa[x]=f; 49 for(int i=linkk[x];i;i=e[i].next){ 50 if(e[i].y==f)continue; 51 dfs(e[i].y,x); 52 siz[x]+=siz[e[i].y]; 53 ru[x]++; 54 } 55 if(siz[x]==1)q[++tail]=x; 56 } 57 void work(){ 58 memset(f,-1,sizeof(f)); 59 findd(); 60 dfs(0,0); 61 for(int head=1;head<=tail;head++){ 62 int x=q[head]; 63 ru[fa[x]]--; 64 if(ru[fa[x]]==0)q[++tail]=fa[x]; 65 LL g[56]; 66 for(int k=0;k<=siz[x]&&k<=K;k++){ 67 for(int prev=fa[x];prev!=-1;prev=fa[prev]){ 68 69 for(int p=0;p<=k;p++)g[p]=0; 70 for(int i=linkk[x];i;i=e[i].next){ 71 if(e[i].y==fa[x])continue; 72 LL minn=inf; 73 for(int j=k;j>=0;j--){ 74 minn=inf; 75 for(int l=0;l<=siz[e[i].y]&&l<=j;l++) 76 minn=min(minn,g[j-l]+f[e[i].y][l][prev]); 77 g[j]=minn; 78 } 79 } 80 LL ans=g[k]+w[x]*d[prev][x]; 81 for(int p=0;p<=k;p++)g[p]=0; 82 for(int i=linkk[x];i;i=e[i].next){ 83 if(e[i].y==fa[x])continue; 84 LL minn=inf; 85 for(int j=k-1;j>=0;j--){ 86 minn=inf; 87 for(int l=0;l<=siz[e[i].y]&&l<=j;l++) 88 minn=min(minn,g[j-l]+f[e[i].y][l][x]); 89 g[j]=minn; 90 } 91 } 92 f[x][k][prev]=min(ans,k-1<0?inf:g[k-1]); 93 if(!prev)break; 94 } 95 } 96 } 97 cout<<f[0][K][0]<<endl; 98 } 99 int main(){100 freopen("1.in","r",stdin);101 freopen("1.out","w",stdout);102 init();103 work();104 }
rivers ioi2005 树形dp
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。