发现对于任意一条边,起决定性作用的是节点编号更大的点.
于是,对于每一条边,按照节点编号较大值作为边权,按照最小生成树的方式插入即可.
最后用线段树维护 dfs 序做一个区间查询即可.
Code:
#include <bits/stdc++.h> #define N 400005 #define ll long long #define mod 998244353 #define setIO(s) freopen(s".in","r",stdin) using namespace std; ll val[N],mul[N<<2],V[N]; int n,m,Q,tot,edges,tt; int p[N],cc[N],hd[N],to[N],nex[N],dfn[N],st[N],ed[N],size[N],Fa[21][N],F[21][N]; struct Edge { int u,v,c; Edge(int u=0,int v=0,int c=0):u(u),v(v),c(c){} }e[N]; void pushup(int l,int r,int now) { ll re=1; int mid=(l+r)>>1; if(l<=mid)re=re*mul[now<<1]%mod; if(r>mid) re=re*mul[now<<1|1]%mod; mul[now]=re; } void build(int l,int r,int now) { if(l==r) { mul[now]=V[l]%mod; return; } int mid=(l+r)>>1; if(l<=mid) build(l,mid,now<<1); if(r>mid) build(mid+1,r,now<<1|1); pushup(l,r,now); } void update(int l,int r,int now,int p,int v) { if(l==r) { V[l]=mul[now]=1ll*v%mod; return; } int mid=(l+r)>>1; if(p<=mid) update(l,mid,now<<1,p,v); else update(mid+1,r,now<<1|1,p,v); pushup(l,r,now); } ll query(int l,int r,int now,int L,int R) { if(l>=L&&r<=R) return mul[now]; ll re=1; int mid=(l+r)>>1; if(L<=mid) re=re*query(l,mid,now<<1,L,R)%mod; if(R>mid) re=re*query(mid+1,r,now<<1|1,L,R)%mod; return re; } int find(int x) { return p[x]==x?x:p[x]=find(p[x]); } bool cmp(Edge a,Edge b) { return a.c<b.c; } void addedge(int u,int v) { nex[++edges]=hd[u],hd[u]=edges,to[edges]=v; } void dfs(int u,int ff) { st[u]=dfn[u]=++tt, size[u]=1, V[dfn[u]]=val[u]; Fa[0][u]=ff, F[0][u]=max(cc[ff], cc[u]); for(int i=1;i<=20;++i) { Fa[i][u]=Fa[i-1][Fa[i-1][u]]; F[i][u]=max(F[i-1][u], F[i-1][Fa[i-1][u]]); } for(int i=hd[u];i;i=nex[i]) { int v=to[i]; dfs(v, u); size[u]+=size[v]; } ed[u]=tt; } int get(int x,int k) { for(int i=20;i>=0;--i) if(Fa[i][x] && F[i][x] <= k) x=Fa[i][x]; return x; } int main() { int i,j,k; // setIO("input"); scanf("%d%d%d",&n,&m,&Q); tot=n; for(i=1;i<N;++i) p[i]=i; for(i=1;i<=n;++i) scanf("%lld",&val[i]),cc[i]=i; for(i=1;i<=m;++i) { scanf("%d%d",&e[i].u,&e[i].v); e[i].c=max(e[i].u, e[i].v); } sort(e+1,e+1+m,cmp); for(i=1;i<=m;++i) { int u=e[i].u, v=e[i].v; int x=find(u),y=find(v); if(x!=y) { ++tot; val[tot]=1ll; cc[tot]=e[i].c; p[x]=tot,p[y]=tot; addedge(tot,x), addedge(tot,y); } } cc[0]=1000000000; dfs(tot,0); build(1,tot,1); for(i=1;i<=Q;++i) { int op; scanf("%d",&op); if(op==1) { int x,y; scanf("%d%d",&x,&y); if(x>y) printf("0\n"); else { int k=get(x,y); int L=st[k], R=ed[k]; printf("%lld\n",query(1,tot,1,L,R)); } } if(op==2) { int x,y; scanf("%d%d",&x,&y); update(1,tot,1,dfn[x],y); } } return 0; }