【题解】Luogu P1038 神经网络 拓扑排序
topsort 板(?)题
显然,题目要求从入度为零的点遍历到出度为零的点,想到用拓排
另如果是输入层,那么u[i]为1或0都没有关系,一定会激活
同时处理出度便于输出
code
1 #include <bits/stdc++.h> 2 using namespace std; 3 namespace gengyf{ 4 #define ll long long 5 const int inf=1e9+7; 6 const int maxn=1e4+10; 7 inline int read(){ 8 int x=0,f=1; 9 char c=getchar(); 10 while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();} 11 while(c>='0'&&c<='9'){x=(x*10)+c-'0';c=getchar();} 12 return x*f; 13 } 14 struct edge{ 15 int to,nxt,w,from; 16 }e[maxn*2]; 17 int head[maxn],cnt,r[maxn],p[maxn]; 18 int n,m,t,c[maxn],u[maxn],fl; 19 inline void add(int from,int to,int w){ 20 e[++cnt].from=from;e[cnt].to=to;e[cnt].w=w; 21 e[cnt].nxt=head[from];head[from]=cnt; 22 ++r[to];++p[from]; 23 } 24 void topsort(){ 25 queue<int>q; 26 int x,v; 27 for(int i=1;i<=n;i++){ 28 if(!r[i])q.push(i); 29 } 30 while(!q.empty()){ 31 x=q.front();q.pop(); 32 for(int i=head[x];i;i=e[i].nxt){ 33 v=e[i].to;--r[v]; 34 if(c[x]>0)c[v]+=c[x]*e[i].w; 35 if(!r[v])q.push(v); 36 } 37 } 38 } 39 int main(){ 40 n=read();m=read(); 41 for(int i=1;i<=n;i++){ 42 c[i]=read();u[i]=read(); 43 if(c[i]==0)c[i]-=u[i]; 44 } 45 for(int i=1;i<=m;i++){ 46 int x,y,z;x=read();y=read();z=read(); 47 add(x,y,z); 48 } 49 topsort(); 50 for(int i=1;i<=n;i++){ 51 if(c[i]>0&&p[i]==0){ 52 printf("%d %d\n",i,c[i]); 53 fl=1; 54 } 55 } 56 if(!fl)printf("NULL\n"); 57 return 0; 58 } 59 } 60 signed main(){ 61 gengyf::main(); 62 return 0; 63 }