CF613D Kingdom and its Cities 虚树树形dp 贪心

LINK：Kingdom and its Cities

发现是一个树上关键点问题所以考虑虚树刚好也有标志(sum kleq 100000)即关键点总数的限制。

首先当k==1时答案显然为0。

然后考虑无解情况容易发现这种情况是两个点同时为关键点那么我们只需要判断是否存在两个点相连的情况就好了。

这个可以在建立虚树时候判断我多此一举了直接标记父亲然后判断父亲是否存在。

接下来考虑如何统计答案。

可以从下往上观察在某个点处统计儿子们的贡献。

假设儿子都是关键点考虑当前点是否是关键点如果是显然都得断开一遍。且向上传递的时候这个点是关键点。

如果当前不是那么如果有>1个儿子那么在当前断开显然是比较优的且向上传递的时候这个点不是关键点、

如果只有一个儿子可以在当前断开不过更优的是可以去上面断开这个结果不会更差。

其实就这三种情况(类似dp 可以发现第一个决策时固定的第二个决策是最优的第三个决策根据不会更差的那套理论可以发现也是最优的。

注意特判有效儿子为0的情况.

code

//#include<bitsstdc++.h>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<queue>
#include<deque>
#include<stack>
#include<vector>
#include<algorithm>
#include<utility>
#include<bitset>
#include<set>
#include<map>
#define ll long long
#define db double
#define INF 1000000000
#define ldb long double
#define pb push_back
#define put_(x) printf("%d ",x);
#define get(x) x=read()
#define gt(x) scanf("%d",&x)
#define gi(x) scanf("%lf",&x)
#define put(x) printf("%d
",x)
#define putl(x) printf("%lld
",x)
#define gc(a) scanf("%s",a+1)
#define rep(p,n,i) for(RE int i=p;i<=n;++i)
#define go(x) for(int i=lin[x],tn=ver[i];i;tn=ver[i=nex[i]])
#define fep(n,p,i) for(RE int i=n;i>=p;--i)
#define vep(p,n,i) for(RE int i=p;i<n;++i)
#define pii pair<int,int>
#define mk make_pair
#define RE register
#define P 1000000007
#define gf(x) scanf("%lf",&x)
#define pf(x) ((x)*(x))
#define uint unsigned long long
#define ui unsigned
#define EPS 1e-8
#define sq sqrt
#define S second
#define F first
using namespace std;
char buf[1<<15],*fs,*ft;
inline char getc()
{
    return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;
}
inline int read()
{
    RE int x=0,f=1;RE char ch=getc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getc();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getc();}
    return x*f;
}
const int MAXN=100010;
int n,Q,len,cnt,ans,top;
int f[MAXN][20],Log[MAXN],dfn[MAXN],d[MAXN];
int lin[MAXN],ver[MAXN<<1],nex[MAXN<<1];
int a[MAXN],vis[MAXN],s[MAXN],v[MAXN];
vector<int>g[MAXN];
inline int cmp(int x,int y){return dfn[x]<dfn[y];}
inline void add(int x,int y)
{
	ver[++len]=y;
	nex[len]=lin[x];
	lin[x]=len;
}
inline void dfs(int x,int fa)
{
	dfn[x]=++cnt;d[x]=d[fa]+1;f[x][0]=fa;
	rep(1,Log[d[x]],i)f[x][i]=f[f[x][i-1]][i-1];
	for(int i=lin[x];i;i=nex[i])
	{
		int tn=ver[i];
		if(tn==fa)continue;
		dfs(tn,x);
	}
}
inline int LCA(int x,int y)
{
	if(d[x]>d[y])swap(x,y);
	fep(Log[d[y]],0,i)if(d[f[y][i]]>=d[x])y=f[y][i];
	if(x==y)return x;
	fep(Log[d[x]],0,i)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];
	return f[x][0];
}
inline void insert(int x)
{
	if(top==1)return s[++top]=x,void();
	int lca=LCA(s[top],x);
	if(s[top]==lca)return s[++top]=x,void();
	while(top>1&&dfn[s[top-1]]>=dfn[lca])
	{
		g[s[top-1]].pb(s[top]);
		--top;
	}
	if(s[top]!=lca)
	{
		g[lca].pb(s[top]);
		s[top]=lca;
	}
	s[++top]=x;
}
inline void dp(int x)
{
	int mark=0;
	vep(0,g[x].size(),i)
	{
		int tn=g[x][i];
		dp(tn);
		if(!v[tn])continue;
		if(v[x])++ans;
		else ++mark;
		v[tn]=0;
	}
	if(!v[x])
	{
		if(mark==1)v[x]=1;
		else ++ans;
	}
	g[x].clear();
}
int main()
{
	//freopen("1.in","r",stdin);
	get(n);
	rep(2,n,i)
	{
		int get(x),get(y);
		add(x,y);add(y,x);
		Log[i]=Log[i>>1]+1;
	}
	dfs(1,0);get(Q);
	rep(1,Q,T)
	{
		int get(k),flag=0;
		rep(1,k,i)get(a[i]),vis[f[a[i]][0]]=1;
		rep(1,k,i)if(vis[a[i]])flag=1;
		rep(1,k,i)vis[f[a[i]][0]]=0;
		if(k==1){puts("0");continue;}
		if(flag){puts("-1");continue;}
		sort(a+1,a+1+k,cmp);v[a[1]]=1;
		s[top=1]=1;if(a[1]!=1)insert(a[1]);
		rep(2,k,i)insert(a[i]),v[a[i]]=1;
		while(top>1)g[s[top-1]].pb(s[top]),--top;
		ans=0;dp(1);put(ans);v[1]=0;
	}
	return 0;
}
</details>

CF613D Kingdom and its Cities 虚树 树形dp 贪心

相关推荐

CF613D Kingdom and its Cities 虚树树形dp 贪心