LibreOJ #3145.「APIO 2019」桥梁 分块

题意

有一个$n$个点$m$条边的无向图，边有边权。有$q$个操作，每个操作会修改一条边的边权，或是询问从点$x$开始，只走权值不小于$y$的边，可以到达多少个点。
$n\le 5\ast 10^4,m,q\le 10^5$

分析

先对操作分块。处理到某一块的时候，先把之前的块的修改都做掉。然后把这一块中没有被修改的边按边权排序。对这一块的询问按权值从大到小处理，每次把权值不小于当前询问值的非修改边加入，然后暴力扫一遍所有的修改边，把满足条件的加入。
用按秩合并的并查集来维护连通性并支持撤销，每处理完一个询问就把加入的修改边撤销。
每次排序的时候可以用归并来降低复杂度。
调整块大小可以做到$O(q\sqrt{mlogn})$，不过貌似块调成$\sqrt{m}$也能过。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>

const int N=100005;

int n,m,q,f[N],s[N],top,stack[N],id[N],bel[N],tim[N],ans[N];
struct edge{int u,v,d,id;}e[N],e1[N],e2[N];
struct data{int op,x,y;}t[N];
bool vis[N];

bool cmp(edge a,edge b)
{
	return a.d>b.d;
}

bool cmpq(int a,int b)
{
	return t[a].y>t[b].y;
}

int find(int x)
{
	if (f[x]==x) return x;
	else return find(f[x]);
}

void merge(int x,int y)
{
	x=find(x);y=find(y);
	if (x==y) return;
	if (s[x]>s[y]) std::swap(x,y);
	stack[++top]=x;
	f[x]=y;s[y]+=s[x];
}

void rtn(int tim)
{
	while (top>tim)
	{
		int x=stack[top--];
		s[f[x]]-=s[x];
		f[x]=x;
	}
}

void merge_sort()
{
	int t1=0,t2=0;
	for (int i=1;i<=m;i++)
		if (!vis[i]) e1[++t1]=e[i];
		else e2[++t2]=e[i],vis[i]=0;
	std::sort(e2+1,e2+t2+1,cmp);
	int p1=1,p2=1,p=1;
	while (p1<=t1&&p2<=t2)
		if (e1[p1].d>=e2[p2].d) e[p++]=e1[p1++];
		else e[p++]=e2[p2++];
	for (int i=p1;i<=t1;i++) e[p++]=e1[i];
	for (int i=p2;i<=t2;i++) e[p++]=e2[i];
	for (int i=1;i<=m;i++) bel[e[i].id]=i;
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=m;i++) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].d),e[i].id=i;
	scanf("%d",&q);
	for (int i=1;i<=q;i++) scanf("%d%d%d",&t[i].op,&t[i].x,&t[i].y);
	int B=sqrt(m);
	if (!B) B=1;
	std::sort(e+1,e+m+1,cmp);
	for (int i=1;i<=m;i++) bel[e[i].id]=i;
	for (int i=1;i<=n;i++) f[i]=i,s[i]=1;
	for (int l=1,r;l<=q;l=r+1)
	{
		r=std::min(q,l+B-1);
		int tot=0,p=1;
		for (int i=l;i<=r;i++)
			if (t[i].op==1) vis[bel[t[i].x]]=1;
			else id[++tot]=i;
		std::sort(id+1,id+tot+1,cmpq);
		for (int i=1;i<=tot;i++)
		{
			while (p<=m&&e[p].d>=t[id[i]].y)
			{
				if (!vis[p]) merge(e[p].u,e[p].v);
				p++;
			}
			int tmp=top;
			for (int j=id[i]-1;j>=l;j--)
				if (t[j].op==1&&tim[t[j].x]!=id[i])
				{
					if (t[j].y>=t[id[i]].y) merge(e[bel[t[j].x]].u,e[bel[t[j].x]].v);
					tim[t[j].x]=id[i];
				}
			for (int j=id[i]+1;j<=r;j++)
				if (t[j].op==1&&tim[t[j].x]!=id[i]&&e[bel[t[j].x]].d>=t[id[i]].y) merge(e[bel[t[j].x]].u,e[bel[t[j].x]].v),tim[t[j].x]=id[i];
			ans[id[i]]=s[find(t[id[i]].x)];
			rtn(tmp);
		}
		rtn(0);
		for (int i=l;i<=r;i++) if (t[i].op==1) e[bel[t[i].x]].d=t[i].y;
		merge_sort();
	}
	for (int i=1;i<=q;i++) if (ans[i]) printf("%d\n",ans[i]);
	return 0;
}