「NOI2021 D1T3 庆典」题解

uoj675

加强:k6×105


暴力

ust 路径上 正图上 su 且反图上 ut

时间复杂度 O((n+m)q)

正解

只关心可达性,不妨 SCC 缩点成 DAG(点带权)。注意到一个奇怪的条件:

对于三座城市 x,y,z,若 xzyz,那么有 xyyx

若存在边 (x,z),(x,y),(y,z),可以删去边 (x,z)。这样每个点的前驱唯一,又有原图弱连通,即为叶向树
实现上有两种做法:拓扑排序,保留每个点的最后一条入边;或从零入度点开始 dfs,优先递归编号大的儿子,保留 dfs 树(利用 tarjan 缩点的性质)

叶向树

  • k=0ts 子树内时答案为深度差
  • k=1:只需要讨论走不走新边,转化为子树和根链求交
  • k=2:仍然可以分类讨论,但非常困难
  • 一般情况:
    叶向树的结构非常简单,只有祖先到后代的唯一路径。对 s,t 和新边端点建虚树,边权为两点间点权和(不含端点),再加入新边(边权为 0)。点边数只有 O(k),暴力即可

时间复杂度 O(nlogn+klogk)

代码很好写,都是板子

#ifndef FT
	#define NDEBUG
#endif
#include <bits/stdc++.h>
#include <bits/extc++.h>
using namespace std; using namespace __gnu_pbds; using namespace __gnu_cxx;
#define For(i,x,y,...) for(int i=x,##__VA_ARGS__;i<=(y);++i)
#define rFor(i,x,y,...) for(int i=x,##__VA_ARGS__;i>=(y);--i)
#define Rep(i,x,y,...) for(int i=x,##__VA_ARGS__;i<(y);++i)
#define pb emplace_back
#define sz(a) int((a).size())
#define all(a) (a).begin(),(a).end()
#define fi first
#define se second
#define mkp make_pair
typedef long long LL; typedef vector<int> Vi; typedef pair<int,int> Pii;
auto ckmax=[](auto &x,auto y) { return x<y ? x=y,true : false; };
auto ckmin=[](auto &x,auto y) { return y<x ? x=y,true : false; };
sfmt19937 mt(chrono::steady_clock::now().time_since_epoch().count());
int rnd(int l,int r) { return uniform_int_distribution<>(l,r)(mt); }
template<typename T=int>T read() { T x; cin>>x; return x; }

const int mod = 998244353;
struct mint {
	int x; mint(int x=0):x(x<0?x+mod:x<mod?x:x-mod){}
	mint(LL y) { y%=mod, x=y<0?y+mod:y; }
	mint& operator += (const mint &y) { x=x+y.x<mod?x+y.x:x+y.x-mod; return *this; }
	mint& operator -= (const mint &y) { x=x<y.x?x-y.x+mod:x-y.x; return *this; }
	mint& operator *= (const mint &y) { x=1ll*x*y.x%mod; return *this; }
	friend mint operator + (mint x,const mint &y) { return x+=y; }
	friend mint operator - (mint x,const mint &y) { return x-=y; }
	friend mint operator * (mint x,const mint &y) { return x*=y; }
};	mint Pow(mint x,LL y=mod-2) { mint z(1);for(;y;y>>=1,x*=x)if(y&1)z*=x;return z; }

const int N = 3e5+5;

namespace T { // 叶向树
int n,rt,ind,val[N],sum[N],dfn[N],st[19][N];
Vi e[N];
int _min(int x,int y) { return dfn[x]<dfn[y] ? x : y; }
void dfs(int u,int fa) {
	dfn[u] = ++ind, st[0][ind] = fa, sum[u] = sum[fa] + val[u];
	for(int v : e[u]) dfs(v,u);
}
int lca(int u,int v) {
	if( u == v ) return u;
	if( (u=dfn[u]) > (v=dfn[v]) ) swap(u,v);
	int k = __lg(v-u++);
	return _min(st[k][u], st[k][v-(1<<k)+1]);
}
void main() {
	// cerr<<"T:\n"; For(i,1,n) for(int j : e[i]) cerr<<"  "<<i<<" "<<j<<'\n';
	dfs(rt,0);
	For(i,1,18) For(j,1,n-(1<<i)+1) st[i][j] = _min(st[i-1][j], st[i-1][j+(1<<i-1)]);
}
}

namespace DAG {
int n,deg[N];
Vi e[N];
void main() {
	// cerr<<"DAG:\n"; For(i,1,n) for(int j : e[i]) cerr<<"  "<<i<<" "<<j<<'\n';
	T::n = n;
	For(i,1,n) for(int j : e[i]) ++deg[j];
	queue<int> q;
	For(i,1,n) if( !deg[i] ) q.emplace(i), T::rt = i;
	assert(sz(q)==1);
	while( sz(q) ) {
		int u = q.front(); q.pop();
		for(int v : e[u]) if( !--deg[v] ) q.emplace(v), T::e[u].pb(v);
	}
}
}

namespace G { // 原图
int n,m,ind,dfn[N],low[N],id[N];
Vi e[N];
void tj(int u) {
	static int t,s[N];
	dfn[u] = low[u] = ++ind, id[u] = -1, s[++t] = u;
	for(int v : e[u])
		if( !dfn[v] ) tj(v), ckmin(low[u], low[v]);
		else if( !~id[v] ) ckmin(low[u], dfn[v]);
	if( dfn[u] == low[u] ) {
		++DAG::n;
		do id[s[t]] = DAG::n, ++T::val[DAG::n]; while( u != s[t--] );
	}
}
void main() {
	For(i,1,n) if( !dfn[i] ) tj(i);
	For(i,1,n) for(int j : e[i]) if( id[i] != id[j] ) DAG::e[id[i]].pb(id[j]);
	// cerr<<"id: "; For(i,1,n) cerr<<id[i]<<" "; cerr<<'\n';
}
}

int mm;
Vi ver;
struct VT {
	int head[N];
	bool vis[N];
	struct { int nxt,to; } e[30];
	void clr() { for(int i : ver) head[i] = vis[i] = 0; }
	void dfs(int u) {
		if( vis[u] ) return; vis[u] = 1;
		for(int i = head[u], v; v = e[i].to, i; i = e[i].nxt) dfs(v);
	}
} g1,g2;
void adde(int x,int y) {
	g1.e[++mm] = {g1.head[x],y}, g1.head[x] = mm;
	g2.e[mm] = {g2.head[y],x}, g2.head[y] = mm;
	// cerr<<"  "<<x<<" "<<y<<" "<<'\n';
}

void MAIN() {
	int q,k; cin>>G::n>>G::m>>q>>k; For(i,1,G::m, x,y) cin>>x>>y, G::e[x].pb(y);
	G::main(), DAG::main(), T::main();
	while( q-- ) {
		// cerr<<"VT:\n";
		int s,t; s = G::id[read()], t = G::id[read()], ver = {s,t};
		For(i,1,k, x,y)
			x = G::id[read()], y = G::id[read()],
			ver.pb(x), ver.pb(y), adde(x,y);
		auto cmp = [](int x,int y){return T::dfn[x]<T::dfn[y];};
		sort(all(ver),cmp), ver.erase(unique(all(ver)),ver.end());
		rFor(i,sz(ver)-1,1) ver.pb(T::lca(ver[i-1],ver[i]));
		ver.pb(T::rt), sort(all(ver),cmp), ver.erase(unique(all(ver)),ver.end());
		Rep(i,1,sz(ver)) adde(T::lca(ver[i-1],ver[i]),ver[i]);
		g1.dfs(s), g2.dfs(t);
		int ans = 0;
		for(int i : ver) if( g1.vis[i] && g2.vis[i] ) ans += T::val[i];
		For(i,k+1,mm) if(int x = g2.e[i].to, y = g1.e[i].to; g1.vis[x] && g2.vis[y] )
			ans += T::sum[T::st[0][T::dfn[y]]] - T::sum[x];
		cout<<ans<<'\n';
		g1.clr(), g2.clr(), mm = 0;
	}
} signed main() {
#ifdef FT
	freopen("in","r",stdin); freopen("out","w",stdout);
#endif
	ios::sync_with_stdio(0);cin.tie(0);
	int lft=1; while( lft-- ) {
		MAIN();
	}
	return 0;
}
posted @   ft61  阅读(23)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· Deepseek官网太卡,教你白嫖阿里云的Deepseek-R1满血版
· 2分钟学会 DeepSeek API,竟然比官方更好用!
· .NET 使用 DeepSeek R1 开发智能 AI 客户端
· DeepSeek本地性能调优
· 一文掌握DeepSeek本地部署+Page Assist浏览器插件+C#接口调用+局域网访问!全攻略
点击右上角即可分享
微信分享提示