1. 程式人生 > >UVA - 1205 Color a Tree

UVA - 1205 Color a Tree

fin .com mes 求一個 ans %d AD alt can

技術分享圖片

大意就是給你一顆樹,每個點有一個權值w[i],求一個排列使得 所有的父親都在兒子前面 並且排列的權值最小。

排列的權值在這裏定義為 Σ i * w[p[i]] ,其中p[i] 是排列第i個位置的元素。

然後我瞎jb胡了一個算法,對於每個子樹維護一個 p[],表示只考慮子樹內的元素的最優排列。顯然我們只要把一個點的所有兒子都合並之後再把這個點放在排列的第一個位置就可以了。(可以證明這些元素再往上走的時候相對位置還是一樣的)

問題是怎麽合並。

其實這就是個dp,合並x和y子樹的時候,f[i][j] 表示 用了x子樹前i個元素,用了y子樹前j個元素,此時到終點的最優值是多少,然後順帶記錄一下方案就好啦。

(昨天+今天調了 3h+ 主要是因為 fill 用了 0x7f 一直沒有察覺233333,導致max一直是127(真是醉了))

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1005;
int a[maxn][maxn],n,w[maxn],f[maxn][maxn];
int to[maxn*2],ne[maxn*2],hd[maxn],num,m;
bool nxt[maxn][maxn];
inline void add(int x,int y){ to[++num]=y,ne[num]=hd[x],hd[x]=num;}

inline void init(){
	memset(hd,0,sizeof(hd)),num=0;
	memset(f,0x7f,sizeof(f));
	memset(a,0,sizeof(a));
}

inline void Merge(int x,int son){
	int b[a[x][0]+a[son][0]+5],hzx[a[x][0]+5],hzs[a[son][0]+5];
    hzx[a[x][0]+1]=0;
    for(int i=a[x][0];i;i--) hzx[i]=hzx[i+1]+a[x][i];
    hzs[a[son][0]+1]=0;
    for(int i=a[son][0];i;i--) hzs[i]=hzs[i+1]+a[son][i];
    
    f[a[x][0]][a[son][0]]=0;
    for(int i=a[x][0];i>=0;i--)
        for(int j=a[son][0];j>=0;j--){
        	if(i<a[x][0]&&f[i+1][j]<f[i][j]) f[i][j]=f[i+1][j],nxt[i][j]=1;
        	if(j<a[son][0]&&f[i][j+1]<f[i][j]) f[i][j]=f[i][j+1],nxt[i][j]=0;
        	f[i][j]+=hzx[i+1]+hzs[j+1];
		}
	
	int now=0,nx=0,ny=0;
	while(++now<=a[x][0]+a[son][0]){
		if(nxt[nx][ny]) nx++,b[now]=a[x][nx];
		else ny++,b[now]=a[son][ny];
    }
	
	for(int i=0;i<=a[x][0];i++) fill(f[i],f[i]+a[son][0]+1,2000000000);
	a[x][0]+=a[son][0];
	for(int i=1;i<=a[x][0];i++) a[x][i]=b[i];
}

void dfs(int x,int fa){
	for(int i=hd[x];i;i=ne[i]) if(to[i]!=fa){
		dfs(to[i],x);
		Merge(x,to[i]);
	}
	
	a[x][0]++;
	for(int i=a[x][0];i>1;i--) a[x][i]=a[x][i-1];
	a[x][1]=w[x];
}

inline void calc(int x){
	int ans=0;
	for(int i=1;i<=n;i++) ans+=i*a[x][i];
	printf("%d\n",ans);
}

int main(){
	int uu,vv,root;
	while(scanf("%d%d",&n,&root)==2&&n&&root){
		init();
	    for(int i=1;i<=n;i++) scanf("%d",w+i);
	    for(int i=1;i<n;i++) scanf("%d%d",&uu,&vv),add(uu,vv),add(vv,uu);
	    dfs(root,-1),calc(root);
	}
	return 0;
}

  

UVA - 1205 Color a Tree