hdu5909TreeCutting——点分治（树形DP转为序列DP）

#include
#include
#include
using namespace std;
const int N&＃61;1005,M&＃61;1025,mod&＃61;1e9&＃43;7;
int T,n,m,w[N],hd[N],xnt,to[N<<1],nxt[N<<1],siz[N],rt,mn;
int dfn[N],tot,sta[N],top,f[N][M],g[N],nt[N],ans[M]; bool vis[N];
int rdn()
{int ret&＃61;0;bool fx&＃61;1;char ch&＃61;getchar();while(ch>&＃39;9&＃39;||ch<&＃39;0&＃39;){if(ch&＃61;&＃61;&＃39;-&＃39;)fx&＃61;0;ch&＃61;getchar();}while(ch>&＃61;&＃39;0&＃39;&&ch<&＃61;&＃39;9&＃39;)ret&＃61;ret*10&＃43;ch-&＃39;0&＃39;,ch&＃61;getchar();return fx?ret:-ret;
}
int Mx(int a,int b){return a>b?a:b;}
int Mn(int a,int b){return aa:b;}
void upd(int &x){x>&＃61;mod?x-&＃61;mod:0;}
void init()
{xnt&＃61;0;memset(hd,0,sizeof hd);memset(ans,0,sizeof ans); memset(vis,0,sizeof vis);
}
void add(int x,int y){to[&＃43;&＃43;xnt]&＃61;y;nxt[xnt]&＃61;hd[x];hd[x]&＃61;xnt;}
void getrt(int cr,int fa,int s)
{siz[cr]&＃61;1; int mx&＃61;0;for(int i&＃61;hd[cr],v;i;i&＃61;nxt[i])if(!vis[v&＃61;to[i]]&&v!&＃61;fa){getrt(v,cr,s);siz[cr]&＃43;&＃61;siz[v];mx&＃61;Mx(mx,siz[v]);}mx&＃61;Mx(mx,s-siz[cr]);if(mxcr;
}
void dfs(int cr,int fa)
{dfn[cr]&＃61;&＃43;&＃43;tot;g[tot]&＃61;w[cr];for(int i&＃61;hd[cr],v;i;i&＃61;nxt[i])if(!vis[v&＃61;to[i]]&&v!&＃61;fa)dfs(v,cr);
}
void dfsx(int cr,int fa,int lst)
{nt[dfn[cr]]&＃61;lst;int l&＃61;top&＃43;1;for(int i&＃61;hd[cr],v;i;i&＃61;nxt[i])if(!vis[v&＃61;to[i]]&&v!&＃61;fa)sta[&＃43;&＃43;top]&＃61;v;int r&＃61;top;for(int i&＃61;hd[cr],v,p0&＃61;l;i;i&＃61;nxt[i])if(!vis[v&＃61;to[i]]&&v!&＃61;fa){dfsx(v,cr,p0&＃61;&＃61;r?lst:dfn[sta[p0&＃43;1]]);p0&＃43;&＃43;;}
}
void solve(int cr,int s)
{vis[cr]&＃61;1;tot&＃61;0;dfs(cr,0);top&＃61;0;dfsx(cr,0,s&＃43;1);for(int i&＃61;1;i<&＃61;s&＃43;1;i&＃43;&＃43;)memset(f[i],0,sizeof f[i]);f[1][0]&＃61;1;for(int i&＃61;1;i<&＃61;s;i&＃43;&＃43;)for(int j&＃61;0;j){if(!f[i][j])continue;f[i&＃43;1][j^g[i]]&＃43;&＃61;f[i][j];upd(f[i&＃43;1][j^g[i]]);f[nt[i]][j]&＃43;&＃61;f[i][j];upd(f[nt[i]][j]);}f[s&＃43;1][0]--;//dec the emptyfor(int j&＃61;0,k&＃61;s&＃43;1;jf[k][j],upd(ans[j]);for(int i&＃61;hd[cr],v,ts;i;i&＃61;nxt[i])if(!vis[v&＃61;to[i]]){ts&＃61;(siz[cr]>siz[v]?siz[v]:s-siz[cr]);mn&＃61;N;getrt(v,cr,ts);solve(rt,ts);}
}
int main()
{T&＃61;rdn();while(T--){n&＃61;rdn();m&＃61;rdn();for(int i&＃61;1;i<&＃61;n;i&＃43;&＃43;)w[i]&＃61;rdn();init();for(int i&＃61;1,u,v;irdn(),add(u,v),add(v,u);mn&＃61;N;getrt(1,0,n);solve(rt,n);for(int i&＃61;0,j&＃61;m-1;i"%d ",ans[i]);printf("%d\n",ans[m-1]);}return 0;
}