萌新求助函数调用50pts
  • 板块学术版
  • 楼主Ice_Kissღ
  • 当前回复7
  • 已保存回复7
  • 发布时间2020/12/12 18:41
  • 上次更新2023/11/5 06:13:07
查看原帖
萌新求助函数调用50pts
307416
Ice_Kissღ楼主2020/12/12 18:41
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
const int mod=998244353;
const int M=1e6+10;
long long a[N],sum[N],mul[N],t[N],v[N],u[N];
struct node
{
    int nxt,to;
}mp[2*M];
int head[N];
int n,m;  int cnt,cmp;
void insert(int u,int v)
{
    mp[++cnt].to=v;
    mp[cnt].nxt=head[u];
    head[u]=cnt;
}
int ru[N],trps[N];
void top()
{
    queue <int> q;
    for(int i=1;i<=m+1;i++)
    {
        if(ru[i]==0)
        {
            q.push(i); trps[++cmp]=i;
        }
    }
    while(!q.empty())
    {   
        int qwq=q.front(); q.pop();
        for(int i=head[qwq];i;i=mp[i].nxt)
        {   
            int y=mp[i].to; ru[y]--;
            if(ru[y]==0) q.push(y),trps[++cmp]=y;
        }
    } return ;
} 
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cin>>m;
    for(int i=1;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)
        {
            int x,y;
            scanf("%d",&x); mul[i]=1;
            for(int j=1;j<=x;j++)
            {
                scanf("%d",&y);
                insert(i,y); ru[y]++;
            }
        }
    }
    int q;
    scanf("%d",&q); sum[m+1]=mul[m+1]=1;
    for(int i=1;i<=q;i++)
    {
        int x; scanf("%d",&x);
        insert(m+1,x); ru[x]++;
    }
    top();
    for(int i=n;i;i--)
    {
        int qwq=trps[i];//cout<<qwq<<endl;
        for(int j=head[qwq];j;j=mp[j].nxt)
        {
            int y=mp[j].to;
            mul[qwq]=mul[qwq]*mul[y]%mod;
        }
    }
    for(int i=1;i<=n;i++)
    {
        int qwq=trps[i]; long long now=1;
        for(int j=head[qwq];j;j=mp[j].nxt)
        {
            int y=mp[j].to;
            sum[y]=(sum[y]+now*sum[qwq]%mod)%mod;
            now=now*mul[y]%mod;
        }
    }
    for(int i=1;i<=n;i++) a[i]=a[i]*mul[m+1]%mod;
    for(int i=1;i<=m+1;i++) if(t[i]==1) a[u[i]]=(a[u[i]]+v[i]*sum[i]%mod)%mod;
    for(int i=1;i<=n;i++) printf("%lld ",a[i]);
    return 0;
}

RT

2020/12/12 18:41
加载中...