Luogu P7077/CSP 2020 函数调用

拓扑好题~

参考blog

我们先单独考虑乘法标记,显然越后面的乘法标记影响越大.
如果我们暴力处理的话,实际上就是把$dfs$顺序求出,复杂度较大.
即便是在树形图上也要$O(qn)$.

考虑如何优化?
我们先利用乘法结合律进行乘法标记的合并.
我们先对 函数调用关系的 $DAG$ 遍历,求出调用$x$会对整个序列乘上多少.
然后,我们从后往前遍历操作序列,就可以得出每一次操作后整个序列乘上多少.

然后,让$a_i$乘上变化量.
接着,$\Delta a_i$=加法标记*调用次数.
显然这个调用次数可以顺着$DAG$递推,只要注意遍历边的时候是倒序即可(邻接边天然倒序访问).

#include<bits/stdc++.h>
#define gc getchar()
#define TP template<class o>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define REP(i,a,b) for(int i=b;i>=a;i--)
#define FOR(i,n) rep(i,1,n)
#define go(x,y) for(int k##x=last[x],y;(y=e[k##x].y);k##x=e[k##x].next)
using namespace std;
typedef long long ll;
const int N=1e5+10,M=1e6+10,mod=998244353;

TP void qr(o &x) {
	char c=gc; int f=1; x=0;
	while(!isdigit(c)){if(c=='-') f=-1; c=gc;}
	while(isdigit(c))x=x*10+c-'0',c=gc;
	x*=f;
}

TP void qw(o x) {
	if(x<0) putchar('-'),x=-x;
	if(x/10) qw(x/10);
	putchar(x%10+'0');
}

TP void pr1(o x) {qw(x); putchar(' ');}
TP void pr2(o x) {qw(x); puts("");}

int n,a[N],m,op[N],pos[N],q,f[N];
ll ad[N],mu[N],cnt[N];
bool v[N];

struct edge{int y,next; } e[M]; int len,last[N];
void ins(int x,int y) {e[++len]=(edge){y,last[x]}; last[x]=len;}

template<class t1,class t2>
void AD(t1 &x,t2 y) {
	x += y;
	if(x >= mod) x -= mod;
}

ll dfs(int x) {
	if(v[x]) return mu[x];
	v[x]=1;
	go(x,y) mu[x]=mu[x]*dfs(y)%mod;
	return mu[x];
}

int deg[N];
void topsort() {
	static int l,r,q[N];
	l=1; r=0;
	FOR(i,m) if(!deg[i]) q[++r]=i;
	while(l<=r) {
		int x=q[l++];
		go(x,y) {
			AD(cnt[y],cnt[x]);
			cnt[x]=cnt[x]*mu[y]%mod;
			if(!--deg[y]) q[++r]=y;
		}
	}
	FOR(i,m) if(op[i]==1) AD(a[pos[i]],cnt[i]*ad[i]%mod);
}

int main() {
	qr(n); FOR(i,n) qr(a[i]);
	qr(m); FOR(i,m) {
		qr(op[i]); mu[i]=1;
		if(op[i] == 1) qr(pos[i]),qr(ad[i]);
		else if(op[i] == 2) qr(mu[i]);
		else {
			int k,x; qr(k); 
			while(k--) {
				qr(x); 
				ins(i,x); deg[x]++;
			}
		}
	}
	FOR(i,m) 
		if(!v[i]) 
			mu[i]=dfs(i);
	qr(q); FOR(i,q) qr(f[i]);
	ll s=1; REP(i,1,q) {
		AD(cnt[f[i]],s);
		s=s*mu[f[i]]%mod;
	}
	FOR(i,n) a[i]=a[i]*s%mod;
	topsort();
	FOR(i,n) pr1(a[i]);
	return 0;
}