NOIP2018 保卫王国(倍增+DP)

15 篇文章 0 订阅

吾曾切树,亦曾卫国

d p i , j , a , b dp_{i,j,a,b} dpi,j,a,b表示 i i i的必选/不必选状态为 a a a, f a i 2 j fa^{2^j}_i fai2j的必选/不必选状态为 b b b
然后就倍增和维护DP。

最小点覆盖(了所有的边) = 最大匹配 = s u m − sum - sum最大独立集 = s u m − sum- sum最小边覆盖(了所有的点,最后一个等号在二分图时成立,树显然是二分图)

所以(我也不知道我为什么要)我们可以直接做最大独立集。

A C   C o d e \rm AC \ Code AC Code

#include<bits/stdc++.h>
#define maxn 100005
#define LL long long
#define inf 0x3f3f3f3f3f3f3f3fll
#define lim 17
using namespace std;

int n,m;char typ[3];
LL f[maxn][2],g[maxn][2],w[maxn],dp[lim][maxn][2][2];
int fa[lim][maxn],se[lim][maxn],dep[maxn];
int info[maxn],Prev[maxn<<1],to[maxn<<1],cnt_e;
void Node(int u,int v){ Prev[++cnt_e]=info[u],info[u]=cnt_e,to[cnt_e]=v; }

void dfs1(int u,int ff){
	dep[u] = dep[fa[0][u] = ff] + 1 , se[0][u] = u;
	f[u][1] += w[u];
	for(int i=info[u],v;i;i=Prev[i])
		if((v=to[i])!=ff){
			dfs1(v,u);
			f[u][0] += max(f[v][0] , f[v][1]);
			f[u][1] += f[v][0];
		}
	dp[0][u][0][0] = dp[0][u][0][1] = f[u][0];
	dp[0][u][1][0] = f[u][1] , dp[0][u][1][1] = -inf;
}
void dfs2(int u,int ff){
	for(int i=info[u],v;i;i=Prev[i])
		if((v=to[i])!=ff){
			g[v][0] = max(g[u][1] + f[u][1] - f[v][0] , g[u][0] + f[u][0] - max(f[v][0] , f[v][1]));
			g[v][1] = g[u][0] + f[u][0] - max(f[v][0] , f[v][1]);
			dfs2(v,u);
		}
}
void merge(LL a[2][2],LL b[2][2],LL r[2][2],int p){
	static LL c[2][2];
	c[0][0] = max(a[0][0] + b[0][0] - max(f[p][0],f[p][1]) , a[0][1] +b[1][0] - f[p][0]);
	c[0][1] = max(a[0][0] + b[0][1] - max(f[p][0],f[p][1]) , a[0][1] +b[1][1] - f[p][0]);
	c[1][0] = max(a[1][0] + b[0][0] - max(f[p][0],f[p][1]) , a[1][1] +b[1][0] - f[p][0]);
	c[1][1] = max(a[1][0] + b[0][1] - max(f[p][0],f[p][1]) , a[1][1] +b[1][1] - f[p][0]);
	r[0][0]=c[0][0],r[1][0]=c[1][0],r[0][1]=c[0][1],r[1][1]=c[1][1];
}

int getk(int u,int k){
	for(int i=0;i<lim;i++)
		if(k>>i&1)
			u=fa[i][u];
	return u;
}

int main(){
	
	freopen("defense.in","r",stdin);
	freopen("defense.out","w",stdout);
	
	scanf("%d%d%s",&n,&m,typ);
	LL sum = 0;
	for(int i=1;i<=n;i++) scanf("%lld",&w[i]),sum+=w[i];
	for(int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),Node(u,v),Node(v,u);
	dfs1(1,0);
	dfs2(1,0);
	for(int j=1;j<lim;j++)
		for(int i=1;i<=n;i++){
			fa[j][i] = fa[j-1][fa[j-1][i]] , se[j][i] = fa[j-1][se[j-1][i]];
			if(fa[j][i]){
				int v = fa[j-1][i] , p = se[j-1][i];
				merge(dp[j-1][i],dp[j-1][v],dp[j][i],p);
			}
		}
	for(int u,v,x,y;m--;){
		scanf("%d%d%d%d",&u,&x,&v,&y);
		if(dep[u] < dep[v]) swap(x,y),swap(u,v);
		x^=1,y^=1;
		LL a[2][2]={{-inf,-inf},{-inf,-inf}},b[2][2]={{-inf,-inf},{-inf,-inf}};int sa=0,sb=0;
		a[x][x] = b[y][y] = 0;
		if(fa[0][u] == v && x && y){
			puts("-1");
			continue;
		}
		for(int i=lim-1;i>=0;i--)
			if(dep[fa[i][u]]>=dep[v]){
				merge(a,dp[i][u],a,sa);
				if(!sa) sa = getk(u,(1<<i)-1);
				else sa = fa[i][sa];
				u = fa[i][u];
			}
		LL ans;
		if(u == v)
			ans = a[x][y] + g[v][y] + f[v][y] - (y ? f[sa][0] : max(f[sa][0] , f[sa][1]));
		else{
			for(int i=lim-1;i>=0;i--)
				if(fa[i][u] != fa[i][v]){
					merge(a,dp[i][u],a,sa);
					merge(b,dp[i][v],b,sb);
					if(!sa) sa = getk(u,(1<<i)-1);
					else sa = fa[i][sa];
					if(!sb) sb = getk(v,(1<<i)-1);
					else sb = fa[i][sb];
					u = fa[i][u] , v = fa[i][v];
				} 
			merge(a,dp[0][u],a,sa);
			merge(b,dp[0][v],b,sb);
			sa = u , sb = v;
			int lca = fa[0][u];
			ans = max(a[x][0]+b[y][0]+g[lca][0]+f[lca][0]-max(f[sa][0],f[sa][1])-max(f[sb][0],f[sb][1]),
				a[x][1]+b[y][1]+g[lca][1]+f[lca][1]-f[sa][0]-f[sb][0]);
		}
		printf("%lld\n",ans < 0 ? -1 : sum - ans);
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值