NC16679 [NOIP2003]神经网络

题目链接

题目

题目描述

人工神经网络(Artificial Neural Network)是一种新兴的具有自我学习能力的计算系统,在模式识别、函数逼近及贷款风险评估等诸多领域有广泛的应用。对神经网络的研究一直是当今的热门方向,兰兰同学在自学了一本神经网络的入门书籍后,提出了一个简化模型,他希望你能帮助他用程序检验这个神经网络模型的实用性。

在兰兰的模型中,神经网络就是一张有向图,图中的节点称为神经元,而且两个神经元之间至多有一条边相连,下图是一个神经元的例子:

img

​ 神经元〔编号为1)

​ 图中,X1—X3是信息输入渠道,Y1-Y2是信息输出渠道,C1表示神经元目前的状态,Ui是阈值,可视为神经元的一个内在参数。神经元按一定的顺序排列,构成整个神经网络。在兰兰的模型之中,神经网络中的神经无分为几层;称为输入层、输出层,和若干个中间层。每层神经元只向下一层的神经元输出信息,只从上一层神经元接受信息。下图是一个简单的三层神经网络的例子。 兰兰规定,Ci服从公式:img(其中n是网络中所有神经元的数目)

img

​ 公式中的Wji(可能为负值)表示连接j号神经元和 i号神经元的边的权值。当 Ci大于0时,该神经元处于兴奋状态,否则就处于平静状态。当神经元处于兴奋状态时,下一秒它会向其他神经元传送信号,信号的强度为Ci。

​ 如此.在输入层神经元被激发之后,整个网络系统就在信息传输的推动下进行运作。现在,给定一个神经网络,及当前输入层神经元的状态(Ci),要求你的程序运算出最后网络输出层的状态。

输入描述

第一行是两个整数n(1≤n≤200)和p。
接下来n行,每行两个整数,第i+1行是神经元i最初状态和其阈值(Ui),非输入层的神经元开始时状态必然为0。
再下面P行,每行由两个整数i,j及一个整数Wij,表示连接神经元i、j的边权值为Wij。

输出描述

包含若干行,每行有两个整数,分别对应一个神经元的编号,及其最后的状态,两个整数间以空格分隔。
仅输出最后状态非零的输出层神经元状态,并且按照编号由小到大顺序输出!
若输出层的神经元最后状态均为0,则输出 NULL。

示例1

输入

5 6
1 0
1 0
0 1
0 1
0 1
1 3 1
1 4 1
1 5 1
2 3 1
2 4 1
2 5 1

输出

3 1
4 1
5 1

题解

知识点:拓扑排序。

显然在拓扑排序的过程中处理权值即可。

注意,权值小于等于 \(0\) 的点不参与计算;入度为 \(0\) 的起点不需要减去阈值直接进队,其他的点进队前减去阈值。

最后只输出出度为 \(0\) 且权值大于 \(0\) 的点的信息。

时间复杂度 \(O(n+m)\)

空间复杂度 \(O(n+m)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

const int N = 507, M = 507 * 507;

int n, p;

struct edge {
    int v, nxt, w;
}e[M];
int h[N], idx, C[N], U[N];
void add(int u, int v, int w) {
    e[++idx] = { v,h[u],w };
    h[u] = idx;
}


int deg[N];
queue<int> q;
void toposort() {
    for (int i = 1;i <= n;i++) if (!deg[i]) q.push(i);
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        for (int i = h[u];i;i = e[i].nxt) {
            int v = e[i].v, w = e[i].w;
            C[v] += w * max(C[u], 0);
            deg[v]--;
            if (!deg[v]) q.push(v), C[v] -= U[v];
        }
    }
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    cin >> n >> p;
    for (int i = 1;i <= n;i++) cin >> C[i] >> U[i];
    for (int i = 1;i <= p;i++) {
        int u, v, w;
        cin >> u >> v >> w;
        add(u, v, w);
        deg[v]++;
    }
    toposort();
    bool ok = 0;
    for (int i = 1;i <= n;i++) {
        if (!h[i] && C[i] > 0) cout << i << ' ' << C[i] << '\n', ok = 1;
    }
    if (!ok) cout << "NULL" << '\n';
    return 0;
}
posted @ 2023-01-02 17:20  空白菌  阅读(20)  评论(0编辑  收藏  举报