描述
函数是各种编程语言中一项重要的概念,借助函数,我们总可以将复杂的任务分解成一个个相对简单的子任务,直到细化为十分简单的基础操作,从而使代码的组织更加严密、更加有条理。然而,过多的函数调用也会导致额外的开销,影响程序的运行效率。
某数据库应用程序提供了若干函数用以维护数据。已知这些函数的功能可分为三类:
- 将数据中的指定元素加上一个值;
- 将数据中的每一个元素乘以一个相同值;
- 依次执行若干次函数调用,保证不会出现递归(即不会直接或间接地调用本身)。
在使用该数据库应用时,用户可一次性输入要调用的函数序列(一个函数可能被调用多次),在依次执行完序列中的函数后,系统中的数据被加以更新。某一天,小 A 在应用该数据库程序处理数据时遇到了困难:由于频繁而低效的函数调用,系统在执行操作时进入了无响应的状态,他只好强制结束了数据库程序。为了计算出正确数据,小 A 查阅了软件的文档,了解到每个函数的具体功能信息,现在他想请你根据这些信息帮他计算出更新后的数据应该是多少。
输入描述
第一行一个正整数 n,表示数据的个数。
第二行 n 个整数,第 i 个整数表示下标为 i 的数据的初始值为 ai。
第三行一个正整数 m,表示数据库应用程序提供的函数个数。函数从 1∼m 编号。
接下来 m 行中,第 j(1≤j≤m)行的第一个整数为 Tj,表示 j 号函数的类型:
- 若 Tj=1,接下来两个整数 Pj,Vj 分别表示要执行加法的元素的下标及其增加的值;
- 若 Tj=2,接下来一个整数 Vj 表示所有元素所乘的值;
- 若 Tj=3,接下来一个正整数 Cj 表示 j 号函数要调用的函数个数,
随后 Cj 个整数 g1(j),g2(j),…,gCj(j) 依次表示其所调用的函数的编号。
第 m+4 行一个正整数 Q,表示输入的函数操作序列长度。
第 m+5 行 Q 个整数 fi,第 i 个整数表示第 i 个执行的函数的编号。
输出描述
一行 n 个用空格隔开的整数,按照下标 1∼n 的顺序,分别输出在执行完输入的函数序列后,数据库中每一个元素的值。答案对 998244353 取模。
用例输入 1
3 1 2 3 3 1 1 1 2 2 3 2 1 2 2 2 3
用例输出 1
6 8 12
用例输入 2
10 1 2 3 4 5 6 7 8 9 10 8 3 2 2 3 3 2 4 5 3 2 5 8 2 2 3 2 6 7 1 2 5 1 7 6 2 3 3 1 2 3
用例输出 2
36 282 108 144 180 216 504 288 324 360
提示
【样例 #1 解释】
1 号函数功能为将 a1 的值加一。2 号函数功能为所有元素乘 2。3 号函数将先调用 1 号函数,再调用 2 号函数。
最终的函数序列先执行 2 号函数,所有元素的值变为 2,4,6。
再执行 3 号函数时,先调用 1 号函数,所有元素的值变为 3,4,6。再调用 2 号函数,所有元素的值变为 6,8,12。
【数据范围】
测试点编号 | n,m,Q≤ | ∑Cj | 其他特殊限制 |
---|---|---|---|
1∼2 | 1000 | =m−1 | 函数调用关系构成一棵树 |
3∼4 | 1000 | ≤100 | 无 |
5∼6 | 20000 | ≤40000 | 不含第 2 类函数或不含第 1 类函数 |
7 | 20000 | =0 | 无 |
8∼9 | 20000 | =m−1 | 函数调用关系构成一棵树 |
10∼11 | 20000 | ≤2×105 | 无 |
12∼13 | 105 | ≤2×105 | 不含第 2 类函数或不含第 1 类函数 |
14 | 105 | =0 | 无 |
15∼16 | 105 | =m−1 | 函数调用关系构成一棵树 |
17∼18 | 105 | ≤5×105 | 无 |
19∼20 | 105 | ≤106 | 无 |
对于所有数据:0≤ai≤104,Tj∈{1,2,3},1≤Pj≤n,0≤Vj≤104,1≤gk(j)≤m,1≤fi≤m。
祝大家AC
#include<iostream>
#include<cstdio>
#include<queue>
#define ll long long
using namespace std;
const int N=1e5+100,M=1e6+100;
const ll P=998244353;
int n,m,Q,tot,cnt;
int b[N],t[N],ind[N];
int head[N],ver[2*M],Next[2*M];
ll ans;
ll a[N],u[N],v[N],sum[N],mul[N];
void add(int x,int y)
{
ver[++tot]=y,Next[tot]=head[x],head[x]=tot,ind[y]++;
}
void topusort()
{
queue<int> q;
for(int i=1;i<=m+1;i++)
if(!ind[i])
q.push(i);
while(q.size())
{
int x=q.front();q.pop();
b[++cnt]=x;
for(int i=head[x];i;i=Next[i])
{
int y=ver[i];ind[y]--;
if(!ind[y])
q.push(y);
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
scanf("%d",&m);
for(int i=1,x,y;i<=m;i++)
{
scanf("%d",&t[i]);
if(t[i]==1)
scanf("%lld%lld",&u[i],&v[i]),mul[i]=1;
if(t[i]==2)
scanf("%lld",&u[i]),mul[i]=u[i];
if(t[i]==3)
{
scanf("%d",&x);mul[i]=1;
for(int j=1;j<=x;j++)
{
scanf("%d",&y);
add(i,y);
}
}
}
scanf("%d",&Q);
for(int i=1,x;i<=Q;i++)
{
scanf("%d",&x);
add(m+1,x);
}
topusort();sum[m+1]=mul[m+1]=1;
for(int i=cnt;i;i--)
{
int x=b[i];
for(int j=head[x];j;j=Next[j])
{
int y=ver[j];
mul[x]=mul[x]*mul[y]%P;
}
}
for(int i=1;i<=cnt;i++)
{
int x=b[i];ll now=1;
for(int j=head[x];j;j=Next[j])
{
int y=ver[j];
sum[y]=(sum[y]+now*sum[x]%P)%P;
now=now*mul[y]%P;
}
}
for(int i=1;i<=n;i++)
a[i]=a[i]*mul[m+1]%P;
for(int i=1;i<=m+1;i++)
if(t[i]==1)
a[u[i]]=(a[u[i]]+v[i]*sum[i]%P)%P;
for(int i=1;i<=n;i++)
printf("%lld ",a[i]);
return 0;
}