题意(by 某轻小说看多了的yhn):
吸血鬼有两种产生方式:
1、被它的主人(一个吸血鬼)“初拥”。
2、被它的父母(两个吸血鬼)所生。但这类吸血鬼很少,最多只有15个
现在,给出每个吸血鬼的主人(或父母),要求选择尽可能少的吸血鬼,使得每个主仆关系、每个父子,母子关系中,都至少有一个被选中。求选择的方案数。
题解:
我们考虑一下树上最小边覆盖(是叫这个东西吧?),这个东西是可以直接用dp做的。用dp[u][0/1]表示以u为根的子树中,是否选u的最小的点数。转移时,对于u的儿子v,
d
p
[
u
]
[
0
]
=
∑
d
p
[
v
]
[
1
]
,
d
p
[
u
]
[
1
]
=
∑
min
{
d
p
[
v
]
[
0
]
,
d
p
[
v
]
[
1
]
}
dp[u][0]=\sum dp[v][1],dp[u][1]=\sum \min\{dp[v][0],dp[v][1]\}
dp[u][0]=∑dp[v][1],dp[u][1]=∑min{dp[v][0],dp[v][1]}。方案数就可以对应地用乘法原理求出。
然而这不是一棵树。
又然而最多只有15个点有两个父亲。
所以就让这些点只连一个父亲,另一个父亲存下这个儿子,然后直接枚举这些点是否选择,就把它当成一棵树跑,然后稍微判断一下即可。
btw,topcoder的提交方式适应了的话也海星。
代码:
#include<cstring>
#include<vector>
#include<algorithm>
#define mod 1000000007
#define maxn 1005
#define INF 23333
using namespace std;
int n,x[maxn],f[maxn][2],g[maxn][2],minx=INF,ans;
struct node { int v; node *nxt; } edge[maxn],*head[maxn],*ncnt;
vector<int> son[maxn],sp;
void addedge(int u,int v)
{
ncnt++;
ncnt->v=v,ncnt->nxt=head[u];
head[u]=ncnt;
}
void dfs(int u)
{
f[u][0]=0,f[u][1]=1,g[u][0]=1,g[u][1]=1;
for(node *p=head[u];p;p=p->nxt)
{
int v=p->v; dfs(v);
f[u][0]+=f[v][1],g[u][0]=1ll*g[u][0]*g[v][1]%mod;
if(f[v][0]<f[v][1]) f[u][1]+=f[v][0],g[u][1]=1ll*g[u][1]*g[v][0]%mod;
else if(f[v][1]<f[v][0]) f[u][1]+=f[v][1],g[u][1]=1ll*g[u][1]*g[v][1]%mod;
else f[u][1]+=f[v][1],g[u][1]=1ll*g[u][1]*(g[v][0]+g[v][1])%mod;
}
for(int i=0,siz=son[u].size();i<siz;i++)
if(!x[son[u][i]]) f[u][0]=INF,g[u][0]=0;
if(!x[u]) f[u][1]=INF,g[u][1]=0;
else if(x[u]==1) f[u][0]=INF,g[u][0]=0;
}
class VampireTreeDiv2
{
public:
int countMinSamples(vector<int> a,vector<int> b)
{
n=a.size(),ncnt=&edge[0];
memset(x,-1,sizeof(x));
for(int i=0;i<n;i++)
if(a[i]<0) addedge(b[i],i+1);
else if(b[i]<0) addedge(a[i],i+1);
else
{
addedge(a[i],i+1);
son[b[i]].push_back(i+1);
sp.push_back(i+1);
}
int m=sp.size();
for(int s=0;s<(1<<m);s++)
{
for(int i=0;i<m;i++) x[sp[i]]=(s>>i)&1;
dfs(0);
if(f[0][0]<minx) minx=f[0][0],ans=g[0][0];
else if(f[0][0]==minx) ans=(ans+g[0][0])%mod;
if(f[0][1]<minx) minx=f[0][1],ans=g[0][1];
else if(f[0][1]==minx) ans=(ans+g[0][1])%mod;
}
return ans;
}
};