BZOJ 4042 Luogu P4757 [CERC2014]Parades (树形DP、状压DP)

题目链接

(BZOJ) https://www.lydsy.com/JudgeOnline/problem.php?id=4042

(Luogu) https://www.luogu.org/problem/P4757

题解

挺神仙的题。

观察到两个重要性质:

(1) 只有不影响任何已选方案的时候,才需要去考虑是否要选择\(u\)的子树内往上走的链。(因为链不带权值)

(2) 如果要选择\(u\)子树内往上走的链,那么最多选择一条。

由此可知,我们可以记录哪些链在\(u\)子树内的所有方案中是必选的,所有非必选的都可视作空闲。因为往上走的链最多选择一条,所以如果这条链和一条非必选的边冲突,有办法调整最优方案使得这条链选上,而不会受到“两条非必选的边至少选一条”这种情况的影响。

于是我们记录\(dp[u]\)表示\(u\)子树内最多选择多少链以及\(S[u]\)表示\(u\)子树内最优解非必选的端点,转移的时候先把所有的儿子拿出来,建一个图,两点\(i,j\)连边当且仅当存在\(x\in S[i],y\in S[j], x,y\)是输入的一条路径,然后用状压DP求它的最大匹配即可。设\(dp0[i]\)表示\(i\)集合内的最大匹配,\(U\)为所有儿子的全集,那么若\(dp0[U]=dp0[U-\{i\}]\)就说明\(i\)非必选,那么把\(S[i]\)加入到\(S[u]\)中。

注意特殊处理以\(u\)为一条链的端点的情况。

时间复杂度\(O(n^2+nd2^d+m)\).

代码

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cassert>
#include<vector>
using namespace std; const int N = 1000;
const int M = 5e5;
const int D = 10;
int lg2[(1<<D)+3];
struct Edge
{
int v,nxt;
} e[(N<<1)+3];
int fe[N+3];
int fa[N+3];
bool a[N+3][N+3];
vector<int> ac[N+3];
int dp[N+3];
vector<int> son;
bool ae[D+3][D+3];
bool ae0[D+3];
int dp0[(1<<D)+3];
int n,en,m; void addedge(int u,int v)
{
en++; e[en].v = v;
e[en].nxt = fe[u]; fe[u] = en;
} void dfs(int u)
{
dp[u] = 0;
for(int i=fe[u]; i; i=e[i].nxt)
{
int v = e[i].v;
if(v==fa[u]) continue;
fa[v] = u;
dfs(v);
dp[u] += dp[v];
}
son.clear();
for(int i=fe[u]; i; i=e[i].nxt)
{
int v = e[i].v;
if(v==fa[u]) continue;
son.push_back(v);
}
for(int i=0; i<son.size(); i++)
{
for(int j=i+1; j<son.size(); j++)
{
ae[i][j] = ae[j][i] = false;
for(int ii=0; ii<ac[son[i]].size(); ii++)
{
for(int jj=0; jj<ac[son[j]].size(); jj++)
{
if(a[ac[son[i]][ii]][ac[son[j]][jj]])
{
ae[i][j] = ae[j][i] = true; break;
}
}
}
}
ae0[i] = false;
for(int ii=0; ii<ac[son[i]].size(); ii++) {if(a[ac[son[i]][ii]][u]) {ae0[i] = true; break;}}
}
dp0[0] = 0;
for(int i=1; i<(1<<son.size()); i++)
{
int x = lg2[i&(-i)];
dp0[i] = dp0[i^(1<<x)];
if(ae0[x]) {dp0[i]++;}
for(int j=0; j<son.size(); j++)
{
if(i&(1<<j))
{
if(ae[x][j]) {dp0[i] = max(dp0[i],dp0[i^(1<<x)^(1<<j)]+1);}
}
}
}
dp[u] += dp0[(1<<son.size())-1];
for(int i=0; i<son.size(); i++)
{
if(dp0[(1<<son.size())-1]==dp0[((1<<son.size())-1)^(1<<i)])
{
for(int j=0; j<ac[son[i]].size(); j++) {ac[u].push_back(ac[son[i]][j]);}
}
}
ac[u].push_back(u);
} int main()
{
for(int i=0; i<=D; i++) lg2[1<<i] = i;
int T; scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=1; i<n; i++)
{
int u,v; scanf("%d%d",&u,&v);
addedge(u,v); addedge(v,u);
}
scanf("%d",&m);
for(int i=1; i<=m; i++)
{
int u,v; scanf("%d%d",&u,&v);
a[u][v] = a[v][u] = 1;
}
dfs(1);
printf("%d\n",dp[1]);
for(int i=1; i<=n; i++) fe[i] = fa[i] = 0,ac[i].clear();
for(int i=1; i<=en; i++) e[i].v = e[i].nxt = 0;
for(int i=1; i<=n; i++) for(int j=1; j<=n; j++) a[i][j] = a[j][i] = 0;
n = en = m = 0;
}
return 0;
}
上一篇:Markdown快速上手指南


下一篇:如何在一个js文件中引入另外的js文件