P1038 神经网络

坑点

• 当一个点的C值大于0时，才能向下传递C * W 的值
• 第一层（输入层）的点如果C大于0可直接向下传递，不需C -= U；

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
int C[maxn],U[maxn],indegree[maxn],head[maxn],ver[maxn*2],edge[maxn*2],ne[maxn*2];
int tot,n,p,flag;
void add(int x,int y,int z){
    ver[++tot]=y,edge[tot]=z;
    ne[tot]=head[x],head[x]=tot;
    indegree[y]++;
}
void toposort(){
    queue<int>Q;
    for(int i=1;i<=n;i++){
        if(indegree[i]==0)
            Q.push(i);
    }
    while(!Q.empty()){
        int x=Q.front();
        Q.pop();
        if(C[x]>0&&!head[x]) {
            flag=1;
            continue;
        }
        for(int i=head[x];i;i=ne[i]){
            int y=ver[i],z=edge[i];
            if(C[x]>0) C[y]+=C[x]*z;
            indegree[y]--;
            if(indegree[y]==0){
                C[y]-=U[y];
                Q.push(y);
            }
        }
    }
}
int main(){
    //freopen("1.txt","r",stdin);
    scanf("%d%d",&n,&p);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&C[i],&U[i]);
    }
    for(int i=0,x,y,z;i<p;i++){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
    }
    toposort();
    if(flag==0) printf("NULL\n");
    else{
        for(int i=1;i<=n;i++){
            if(C[i]>0&&!head[i])
                printf("%d %d\n",i,C[i]);
        }
    }
    return 0;
}

 

• 点赞
• 收藏
• 分享
• • 文章举报

策马奔腾向前冲 发布了134 篇原创文章 · 获赞 6 · 访问量 8143 私信 关注

标签：head,P1038,int,indegree,ne,tot,神经网络,maxn
来源： https://blog.csdn.net/qq_40597059/article/details/104135117