【题解】Luogu P1038 神经网络 拓扑排序

topsort 板(?)题

显然，题目要求从入度为零的点遍历到出度为零的点，想到用拓排

另如果是输入层，那么u[i]为1或0都没有关系，一定会激活

同时处理出度便于输出

code

#include <bits/stdc++.h>
using namespace std;
namespace gengyf{
#define ll long long
const int inf=1e9+7;
const int maxn=1e4+10;
inline int read(){
    int x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=(x*10)+c-'0';c=getchar();}
    return x*f;
}
struct edge{
    int to,nxt,w,from;
}e[maxn*2];
int head[maxn],cnt,r[maxn],p[maxn];
int n,m,t,c[maxn],u[maxn],fl;
inline void add(int from,int to,int w){
    e[++cnt].from=from;e[cnt].to=to;e[cnt].w=w;
    e[cnt].nxt=head[from];head[from]=cnt;
    ++r[to];++p[from];
}
void topsort(){
    queue<int>q;
    int x,v;
    for(int i=1;i<=n;i++){
        if(!r[i])q.push(i);
    }
    while(!q.empty()){
        x=q.front();q.pop();
        for(int i=head[x];i;i=e[i].nxt){
            v=e[i].to;--r[v];
            if(c[x]>0)c[v]+=c[x]*e[i].w;
            if(!r[v])q.push(v);
        }
    }
}
int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++){
        c[i]=read();u[i]=read();
        if(c[i]==0)c[i]-=u[i];
    }
    for(int i=1;i<=m;i++){
        int x,y,z;x=read();y=read();z=read();
        add(x,y,z);
    }
    topsort();
    for(int i=1;i<=n;i++){
        if(c[i]>0&&p[i]==0){
            printf("%d %d
",i,c[i]);
            fl=1;
        }
    }
    if(!fl)printf("NULL
");
    return 0;
}
}
signed main(){
  gengyf::main();
  return 0;
}