首页 > 其他分享 >【THUWC2020】Day2T2(dfs树,DP,线段树合并)

【THUWC2020】Day2T2(dfs树,DP,线段树合并)

时间:2022-10-29 18:56:55浏览次数:45  
标签:ch min int lca THUWC2020 Day2T2 dfs ans dp

对于每一个点 \(u\),我们先找到满足右述条件的深度最小的 \(u\) 祖先 \(f\) 并记这个深度最小的祖先的深度为 \(dp(u)\):\(f\) 能只通过除了树上 \([f,u]\) 路径所包含的边之外的边到达 \(u\)。

那么显然,一次询问 \([a,b]\) 中,对于 \(b\) 的子树中的一点 \(u\),\(1\) 号点能到达 \(u\) 当且仅当 \(\min\limits_{v\in[b,u]} dp(v)\leq dep(a)\)。

我们先看如何把 \(dp(u)\) 给求出来。

我们枚举满足条件的深度最小的祖先 \(f\) 最后一步是通过哪一个点到达 \(u\) 的,也就是枚举所有指向 \(u\) 的且的非树边,设这条边为 \((v,u)\):

  • 若 \((v,u)\) 为后向边,那么 \(v\) 为 \(u\) 的祖先,那么 \(v\) 的上一步仍不能是树边,于是更新:\(dp(u)\gets\min(dp(u),dp(v))\)。

  • 若 \((v,u)\) 为横叉边,设 \(v,u\) 的最近公共祖先为 \(lca\):

    在这里插入图片描述

    我们考虑枚举 \(v\) 的上一步是沿树边从哪里走到了 \(v\):对于树上 \([lca,v]\) 路径上的任意一点 \(w\),我们都可以更新:\(dp(u)\gets \min(dp(u),dp(w))\)。

    这样更新是正确的,因为 \(dp(w)\) 要求不经过树上 \([f,w]\) 路径上的边,那么也就不经过树上 \([f,lca]\) 路径上的边,同时由于这是一棵 dfs 树,既然存在了边 \((v,u)\),也就不存在 \((lca,u]\) 中某一点指向 \((lca,v]\) 中某一点的边,所以这条从 \(f\) 到达 \(w\) 的路径也不会经过 \((lca,u]\)。

    这样更新显然也是充分的,我们充分地枚举了 \(v\) 的上一步沿树边走是从哪里走到 \(v\)。

于是我们考虑拓扑排序进行 DP,至于横叉边转移中的 \(\min\limits_{w\in[lca,v]}dp(w)\) 如何求,我们用倍增即可,因为这里没有修改。我们只需要在拓扑排序中更新完 \(dp(u)\) 时把 \(u\) 往上的倍增数组求出来。

回到询问,现在的问题是多次询问 \([a,b]\),询问 \(b\) 的子树中有多少个点 \(v\) 满足 \(\min\limits_{v\in[b,u]} dp(v)\leq dep(a)\)。

考虑离线,并对 \(b\) 跑线段树合并:

我们在原树上 dfs,并对于每一个当前点 \(b\) 维护这么一棵线段树:下标为 \(i\) 的叶子节点中存储着 \(\min\limits_{v\in[b,u]} dp(v)=i\) 的 \(u\) 的个数,也就是 \(\min\limits_{v\in[b,u]} dp(v)\) 的值域线段树。

然后回溯时合并儿子的线段树再用当前点的 \(dp\) 值更新即可。

碰到询问时直接查询 \(\min\limits_{v\in[b,u]} dp(v)\leq dep(a)\) 的 \(u\) 的个数即可。

#include<bits/stdc++.h>

#define LN 17
#define N 100010
#define INF 0x7fffffff
#define lc(u) ch[u][0]
#define rc(u) ch[u][1]

using namespace std;

inline int read()
{
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9')
	{
		x=(x<<1)+(x<<3)+(ch^'0');
		ch=getchar();
	}
	return x*f;
}

struct Query
{
	int a,id;
	Query(){};
	Query(int x,int y){a=x,id=y;}
};

int n,m,q;
int du[N];
int cnt,head[N],nxt[N],to[N];
int fa[N][LN],d[N],size[N];
int dp[N],minn[N][LN];
int node,rt[N],ch[N*LN][2],sum[N*LN];
int ans[N];

vector<int>e[N];
vector<Query>p[N];

void adde(int u,int v)
{
	to[++cnt]=v;
	nxt[cnt]=head[u];
	head[u]=cnt;
}

void dfs(int u)
{
	size[u]=1;
	for(int i=1;i<=16;i++)
		fa[u][i]=fa[fa[u][i-1]][i-1];
	for(int v:e[u])
	{
		if(fa[v][0]) continue;
		fa[v][0]=u;
		d[v]=d[u]+1;
		adde(u,v);
		dfs(v);
		size[u]+=size[v];
	}
}

int calc(int a,int b)
{
	bool flag=0;
	if(d[a]<d[b]) swap(a,b),flag=1;
	int ans=INF;
	for(int i=16;i>=0;i--)
	{
		if(d[fa[a][i]]>=d[b])
		{
			if(!flag) ans=min(ans,minn[a][i]);
			a=fa[a][i];
		}
	}
	if(a==b) return min(ans,minn[a][0]);
	if(flag) swap(a,b);
	for(int i=16;i>=0;i--)
	{
		if(fa[a][i]!=fa[b][i])
		{
			ans=min(ans,minn[a][i]);
			a=fa[a][i],b=fa[b][i];
		}
	}
	return min(ans,minn[a][1]);
}

void DP()
{
	for(int i=1;i<=n;i++) minn[i][0]=d[i];
	queue<int>q;
	q.push(1);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=1;i<=16;i++)
			minn[u][i]=min(minn[u][i-1],minn[fa[u][i-1]][i-1]);
		for(int v:e[u])
		{
			if(u!=fa[v][0]) minn[v][0]=min(minn[v][0],calc(u,v));
			du[v]--;
			if(!du[v]) q.push(v);
		}
	}
}

void up(int u)
{
	sum[u]=sum[lc(u)]+sum[rc(u)];
}

void merge(int &a,int b,int l,int r)
{
	if(!a)
	{
		a=b;
		return;
	}
	if(!b) return;
	if(l==r)
	{
		sum[a]+=sum[b];
		return;
	}
	int mid=(l+r)>>1;
	merge(lc(a),lc(b),l,mid);
	merge(rc(a),rc(b),mid+1,r);
	up(a);
}

int query(int u,int l,int r,int ql,int qr)
{
	if(!u) return 0;
	if(ql<=l&&r<=qr) return sum[u];
	int mid=(l+r)>>1,ans=0;
	if(ql<=mid) ans+=query(lc(u),l,mid,ql,qr);
	if(qr>mid) ans+=query(rc(u),mid+1,r,ql,qr);
	return ans;
}

void del(int &u,int l,int r,int ql,int qr)
{
	if(!u) return;
	if(ql<=l&&r<=qr)
	{
		u=0;
		return;
	}
	int mid=(l+r)>>1;
	if(ql<=mid) del(lc(u),l,mid,ql,qr);
	if(qr>mid) del(rc(u),mid+1,r,ql,qr);
	up(u);
}

void update(int &u,int l,int r,int x,int v)
{
	if(!u) u=++node;
	if(l==r)
	{
		sum[u]+=v;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid) update(lc(u),l,mid,x,v);
	else update(rc(u),mid+1,r,x,v);
	up(u);
}

void solve(int u)
{
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		solve(v);
		merge(rt[u],rt[v],1,n);
	}
	int nv=minn[u][0];
	int tmp=query(rt[u],1,n,nv+1,n);
	del(rt[u],1,n,nv+1,n);
	update(rt[u],1,n,nv,tmp+1);
	for(Query now:p[u])
		ans[now.id]=size[u]-query(rt[u],1,n,1,d[now.a]);
}

int main()
{
	n=read(),m=read(),q=read();
	for(int i=1;i<=m;i++)
	{
		int u=read(),v=read();
		e[u].push_back(v);
		du[v]++;
	}
	for(int i=1;i<=n;i++)
		sort(e[i].begin(),e[i].end());
	d[1]=1;
	dfs(1);
	DP();
	for(int i=1;i<=q;i++)
	{
		int a=read(),b=read();
		p[b].push_back(Query(a,i));
	}
	solve(1);
	for(int i=1;i<=q;i++)
		printf("%d\n",ans[i]);
	return 0;
}

标签:ch,min,int,lca,THUWC2020,Day2T2,dfs,ans,dp
From: https://www.cnblogs.com/ez-lcw/p/16839395.html

相关文章

  • HDFS的 Shell操作-API操作
    HDFS的Shell操作(重点)2.1基本语法hadoopfs具体命令 OR hdfsdfs具体命令两个是完全相同的。2.2命令大全[[email protected]]$bin/hadoopfs[......
  • HDFS概述
    HDFS概述1.1HDFS产出背景及定义1)HDFS产生背景随着数据量越来越大,在一个操作系统存不下所有的数据,那么就分配到更多的操作系统管理的磁盘中,但是不方便管理和维护,迫切需......
  • 学习心得 HDFS读数据过程
    HDFS读数据过程   第一步:打开文件。用Fliesystem先申明一个对象,然后生成一个子类DistributedFileSystem,这个时候生成FS的实例对象,其实是分布式文件系统HDFS的实例对象......
  • 分布式文件系统HDFS
      ......
  • Hadoop - hdfs源码体系结构
    ......
  • 深度优先搜索——DFS 模板
    voiddfs()//参数用来表示状态{if(到达终点状态){...//根据题意添加return;}if(越界或者是不合法状态)......
  • *PAT_甲级_1053 Path of Equal Weight (30分) (C++)【数组排序/DFS】
    目录​​1,题目描述​​​​ 题目大意​​​​输入​​​​输出​​​​2,思路​​​​数据结构 ​​​​如何排序 ​​​​如何设计DFS算法​​​​3,心路历程​​​​4,代......
  • 整理牛客网---阿里校招笔试后端Java版,dfs和算法题。
    一、2021(校招)阿里巴巴7.22笔试(Java版)1.1题目1给定一个n,求[1,n]这n个数字的排列组合有多少个。条件:相邻的两个数字的绝对值不能等于1.例如:4[2,4,1,3][3,1,4......
  • HDFS基础学习
    HDFS简介HDFS即HadoopDistributedFileSystem,是一个分布式文件系统,用于存储海量数据。一个HDFS集群由一个NameNode和多个DataNode组成。HDFS特性主从架构分块存储......
  • DFS(贪心问题)--解决最少数量装箱问题
    题目描述有重量分别为3,5,7公斤的三种货物,和一个载重量为X公斤的箱子(不考虑体积等其它因素,只计算重量)需要向箱子内装满X公斤的货物,要求使用的货物个数尽可能少(三种货物数量无......