题解：P11448 「ALFR Round 3」D 核裂变

题目传送门

思路

容易发现，一个原子被另一个原子释放的中子轰击过后，就会一直被这个原子释放的中子轰击直到结束为止，所以当原子 $i$ 释放的中子在时刻 $x$ 轰击了原子 $j$，那原子 $j$ 所释放的总能量会增加 $k-x+1$。

我们又观察到，一个原子每秒有可能被多个原子轰击，但对于每个原子最多只会释放出一个中子进行轰击，所以一个原子释放对另一个原子所释放的中子数量不会超过 $1$。

设 $s_i$ 为原子 $i$ 开始被轰击时的时间，$an{s}_i$ 为原子 $i$ 总共释放的能量。我们可以通过 bfs 来对 $s_i$ 进行赋值并累加原子所释放的能量。即当我们遍历到原子 $u$ 可以撞击原子 $v$ 时，如果 $v$ 还没有被撞击过，则将 $s_v$ 赋值为 $s_u+1$，将 $an{s}_v$ 赋值为 $a_v\times (k-s_u)$，并将 $v$ 加入队列中。进行完判断后，再将 $an{s}_v$ 加上原子 $u$ 对原子 $v$ 造成的贡献 $k-s_u$。

不过，$m$ 有可能大于 $1$，所以我们要化零为整，定义一个原子 $0$ 连接 $v_i$，并且 $a_0=m,s_0=0,a_{0,i}=v_i$。

最后输出 $an{s}_i$ 即可。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=5e5+5,mod=998244353;
int n,a[N];
long long k,ans[N],s[N];
vector<int> g[N];
int main(){
	scanf("%d%lld",&n,&k);
	for(int i=0,x;i<=n;i++){
		scanf("%d",&a[i]);
		for(int j=0;j<a[i];j++){
			scanf("%d",&x);
			g[i].push_back(x);
		}
	}
	queue<int> q;
	q.push(0);
	while(q.size()){
		int f=q.front();
		q.pop();
		if(s[f]==k)
			break;//如果s[f]==k，说明队列中的原子无法再释放中子，跳出循环
		for(int i=0;i<a[f];i++){
			int v=g[f][i];
			if(!s[v])
				ans[v]=1LL*a[v]*((k-s[f])%mod)%mod,s[v]=s[f]+1,q.push(v);
			ans[v]+=k-s[f],ans[v]%=mod;//注意取模
		}
	}
	for(int i=1;i<=n;i++)
		printf("%lld ",ans[i]);
	return 0;
}