萌新求助函数调用
  • 板块学术版
  • 楼主Ice_Kissღ
  • 当前回复6
  • 已保存回复6
  • 发布时间2020/12/12 17:49
  • 上次更新2023/11/5 06:13:17
查看原帖
萌新求助函数调用
307416
Ice_Kissღ楼主2020/12/12 17:49
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
const int mod=998244353;
const int M=1e6+10;
int 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;
}

过了样例但全WA(呜呜呜

2020/12/12 17:49
加载中...