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 ;
}