51nod 1673 树有几多愁——虚树+状压DP

题目:http://www.51nod.com/Challenge/Problem.html#!#problemId=1673

建一个虚树。

一种贪心的想法是把较小的值填到叶子上,这样一个小值限制到的叶子比较少。

但不太会贪心了,所以考虑 DP 。只有 20 个叶子,(不是用来暴搜的!)可以状压DP了。

dp[ S ]表示选了点集 S 的叶子的方案数。再记一个 ct[ S ] 表示选这个点集的叶子、不影响到其他叶子,最多可以填几个点。

dp[ S ]可以枚举最后一个填的是哪个叶子来转移;ct[ S ]可以在第一次枚举的时候就算出来,就是看这个 “最后一个填上的叶子” 多带来几个可以填的点。

不过这样就不知道复杂度是怎样的了。反正还是过了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define db double
using namespace std;
int rdn()
{
int ret=;bool fx=;char ch=getchar();
while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
return fx?ret:-ret;
}
ll Mx(ll a,ll b){return a>b?a:b;}
ll Mn(ll a,ll b){return a<b?a:b;} const int N=1e5+,K=,M=(<<)+,mod=1e9+;
int n,hd[N],xnt,to[N<<],nxt[N<<],dfn[N],tim;
int bin[K],lg[M],dep[N],pre[N][K],siz[N]; bool vis[N]; int get_lca(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
int d=dep[x]-dep[y];
for(int t=;bin[t]<=d;t++)
if(d&bin[t])x=pre[x][t];
if(x==y)return x;
for(int t=;t>=;t--)
if(pre[x][t]!=pre[y][t])
x=pre[x][t], y=pre[y][t];
return pre[x][];
}
namespace Tr{
const int tN=;
int hd[N],xnt,to[tN],nxt[tN],fa[N];
int sta[tN],top,p[tN],tot;
int bh[N],dy[tN],vl[N],ct[M];
int dp[M]; db d2[M]; bool cmp(int u,int v){return dfn[u]<dfn[v];}
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;fa[y]=x;}
void ini_dfs(int cr)
{
if(vis[cr])vl[cr]=bin[bh[cr]-];
for(int i=hd[cr],v;i;i=nxt[i])
{
ini_dfs(v=to[i]);
vl[cr]|=vl[v];
}
}
void init()
{
sort(p+,p+tot+,cmp);
sta[top=]=p[];
for(int i=;i<=tot;i++)
{
int u=p[i], lca=get_lca(u,sta[top]);
while(top&&dfn[lca]<dfn[sta[top]])
{
if(dfn[sta[top-]]<dfn[lca])
add(lca,sta[top]);
else add(sta[top-],sta[top]);
top--;
}
if(sta[top]!=lca)sta[++top]=lca;
sta[++top]=u;
}
for(int i=;i<top;i++)add(sta[i],sta[i+]);
ini_dfs(sta[]);
}
void solve()
{
init();
for(int i=,S;i<=tot;i++)
{
S=bin[i-]; dp[S]=; d2[S]=;
int cr=dy[i];//dy not bh
while(vl[fa[cr]]==S)cr=fa[cr];
ct[S]=siz[cr]+(dep[cr]-dep[fa[cr]]-);
}
for(int S=;S<bin[tot];S++)
{
if(lg[S])continue;
int T=(S&-S), cr=dy[lg[T]+], pr=cr; T=S^T;
dp[S]=(ll)dp[T]*(ct[T]+)%mod;
d2[S]=d2[T]*(ct[T]+);
while(cr&&(vl[cr]|S)==S)//token!!
cr=fa[cr];
// if(!cr) then ct[S] is wrong but has no influence
ct[S]=ct[T]+dep[pr]-dep[cr];
int tp=T;
while(tp)
{
T=(tp&-tp); tp^=T; T=S^T;
db tmp=d2[T]*(ct[T]+);
if(tmp>d2[S])
dp[S]=(ll)dp[T]*(ct[T]+)%mod,
d2[S]=d2[T]*(ct[T]+);
}
}
printf("%d\n",dp[bin[tot]-]);
}
}
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
void dfs(int cr,int fa)
{
dfn[cr]=++tim;
siz[cr]=; dep[cr]=dep[fa]+;
pre[cr][]=fa;
for(int t=;bin[t]<=dep[cr];t++)
pre[cr][t]=pre[pre[cr][t-]][t-];
bool flag=;
for(int i=hd[cr],v;i;i=nxt[i])
if((v=to[i])!=fa)
{
flag=;
dfs(v,cr);siz[cr]+=siz[v];
}
if(!flag)
{
vis[cr]=; Tr::p[++Tr::tot]=cr;
Tr::bh[cr]=Tr::tot;
Tr::dy[Tr::tot]=cr;
}
}
int main()
{
n=rdn();
for(int i=,u,v;i<n;i++)
u=rdn(),v=rdn(),add(u,v),add(v,u);
bin[]=;
for(int i=;i<=;i++)
bin[i]=bin[i-]<<,lg[bin[i]]=i;
dfs(,); Tr::solve();
return ;
}
上一篇:大数据技术hadoop入门理论系列之二—HDFS架构简介


下一篇:手写一个关于title属性自定义提示框解决浏览器(IE)不兼容问题