题目
给你一个有向图,图中有 n
个节点,节点编号从 0
到 n - 1
,其中每个节点都 恰有一条 出边。
图由一个下标从 0 开始、长度为 n
的整数数组 edges
表示,其中 edges[i]
表示存在一条从节点 i
到节点 edges[i]
的 有向 边。
节点 i
的 边积分 定义为:所有存在一条指向节点 i
的边的节点的 编号 总和。
返回 边积分 最高的节点。如果多个节点的 边积分 相同,返回编号 最小 的那个。
示例 1:
输入:edges = [1,0,0,0,0,7,7,5]
输出:7
解释:
- 节点 1、2、3 和 4 都有指向节点 0 的边,节点 0 的边积分等于 1 + 2 + 3 + 4 = 10 。
- 节点 0 有一条指向节点 1 的边,节点 1 的边积分等于 0 。
- 节点 7 有一条指向节点 5 的边,节点 5 的边积分等于 7 。
- 节点 5 和 6 都有指向节点 7 的边,节点 7 的边积分等于 5 + 6 = 11 。
节点 7 的边积分最高,所以返回 7 。
示例 2:
输入:edges = [2,0,0,2]
输出:0
解释:
- 节点 1 和 2 都有指向节点 0 的边,节点 0 的边积分等于 1 + 2 = 3 。
- 节点 0 和 3 都有指向节点 2 的边,节点 2 的边积分等于 0 + 3 = 3 。
节点 0 和 2 的边积分都是 3 。由于节点 0 的编号更小,返回 0 。
提示:
n == edges.length
2 <= n <= 10^5
0 <= edges[i] < n
edges[i] != i
解题
方法一:模拟
思路
这题虽然提到了图,但和图没有一丁点关系,直接模拟即可。
维护一个长度为 n
的数组(edgeScores
)当作哈希表用,edgeScores[i]
存的是节点 i
的边积分 ,遍历一遍所有边维护好所有边积分然后遍历数组取一个元素最大的中索引最小的索引即可。
代码
class Solution {
public int edgeScore(int[] edges) {
int n = edges.length;
long[] edgeScores = new long[n];
for (int i = 0; i < n; ++i) edgeScores[edges[i]] += i;
int maxIdx = 0;
for (int i = 1; i < n; ++i) {
if (edgeScores[i] > edgeScores[maxIdx]) maxIdx = i;
}
return maxIdx;
}
}
class Solution {
public:
int edgeScore(vector<int>& edges) {
int n = edges.size();
vector<long long> edge_scores(n, 0);
for (int i = 0; i < n; ++i) edge_scores[edges[i]] += i;
int max_idx = 0;
for (int i = 1; i < n; ++i) {
if (edge_scores[i] > edge_scores[max_idx]) max_idx = i;
}
return max_idx;
}
};
评论区