Description
给出一棵n个节点的树,你需要删去并加入一条边,使得原图仍然是一棵树,并且有完美匹配。
求方案数。
n<=5*1e5
Solution
考虑枚举删去一条边,我们只需要统计某个子树内和外有多少个点可以成为匹配点。
可以设Dp,四种状态,根节点是否被匹配,除根外是否有节点未被匹配。
这样子可以O(n)统计出子树内的答案,但是子树外的答案似乎没有那么好求。
观察我们的转移,是从儿子向父亲合并的,这也限制了我们在换根的时候不能用前缀和来优化。
真的是这样吗?
Orz wxh
我们可以考虑把转移写成一种类似矩阵乘法(当然并不是),使得这个转移满足交换律,并且和谁是父亲无关,只需要把不是父亲的点的trans状态都加起来就是答案。
这个加法是重定义之后的,这样子就可以用前缀和优化换根到O(n)解决问题。
/注释部分是原来的转移/
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a) for(int i=lst[a];i;i=nxt[i])
using namespace std;
typedef long long ll;
int read() {
char ch;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
int x=ch-'0';
for(ch=getchar();ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x;
}
const int N=1e6+5;
int t[N<<1],nxt[N<<1],lst[N],l;
void add(int x,int y) {
t[++l]=y;nxt[l]=lst[x];lst[x]=l;
}
int n,x,y,size[N],a[N],L[N],R[N],tot;
struct Dp{
int a[2][2];
Dp trans() {
Dp x;
x.a[0][0]=a[1][0];
x.a[0][1]=a[0][0]+a[1][1];
x.a[1][0]=a[0][0];
x.a[1][1]=a[0][1];
return x;
}
friend Dp operator + (Dp x, Dp y) {
Dp z;
memset(z.a,0,sizeof(z.a));
fo(i,0,1)
fo(j,0,1)
fo(k,0,1-i)
fo(l,0,1-j)
z.a[i+k][j+l]+=x.a[i][j]*y.a[k][l];
/*z.a[0][0]=x.a[0][0]*y.a[1][0];
z.a[0][1]=x.a[0][0]*(y.a[0][0]+y.a[1][1]);
z.a[0][1]+=x.a[0][1]*y.a[1][0];
z.a[1][0]=x.a[0][0]*y.a[0][0];
z.a[1][0]+=x.a[1][0]*y.a[1][0];
z.a[1][1]=x.a[0][0]*y.a[0][1];
z.a[1][1]+=x.a[0][1]*y.a[0][0];
z.a[1][1]+=x.a[1][0]*(y.a[0][0]+y.a[1][1]);
z.a[1][1]+=x.a[1][1]*y.a[1][0];*/
return z;
}
}f[N],g[N],pre[N],Null;
void dfs(int x,int y) {
rep(i,x) if (t[i]!=y) dfs(t[i],x);
f[x]=Null;size[x]=1;
rep(i,x)
if (t[i]!=y) {
size[x]+=size[t[i]];
f[x]=f[x]+f[t[i]].trans();
}
}
void dp(int x,int y) {
L[x]=++tot;
rep(i,x) if (t[i]!=y) a[++tot]=t[i];
R[x]=tot;
pre[L[x]]=y?g[x].trans():Null;
fo(i,L[x]+1,R[x]) pre[i]=pre[i-1]+f[a[i]].trans();
Dp now=Null;
fd(i,R[x],L[x]+1) {
g[a[i]]=pre[i-1]+now;
now=now+f[a[i]].trans();
}
fo(i,L[x]+1,R[x]) dp(a[i],x);
}
int main() {
n=read();
fo(i,1,n-1) {
x=read();y=read();
add(x,y);add(y,x);
}
Null.a[0][0]=1;
dfs(1,0);dp(1,0);
ll ans=0;
fo(i,2,n)
if (f[i].a[1][0]==1&&g[i].a[1][0]==1) ans+=(ll)size[i]*(n-size[i]);
else ans+=(ll)(f[i].a[0][0]+f[i].a[1][1])*(g[i].a[0][0]+g[i].a[1][1]);
printf("%lld\n",ans);
return 0;
}