C++中所有人成为朋友的最早时刻


假设在一个社交群体中,有N个不同的人,其唯一的整数ID从0到N-1。这里我们有一份日志列表,其中每个logs[i] = [时间,id_A,id_B]包含一个非负整数时间戳,以及两个不同人的ID。每个日志都显示了两个人成为朋友的时间。如果A与B是朋友,那么B也与A是朋友。假设如果A与B是朋友,或者A是与B认识的人的朋友,那么A认识B。我们必须找到每个每个人都认识其他每个人的最早时间。如果没有找到这样的时间,则返回-1。例如,如果输入如下:[[20190101,0,1], [20190104,3,4], [20190107,2,3], [20190211,1,5], [20190224,2,4], [20190301,0,3], [20190312,1,2], [20190322,4,5]],N = 6,输出将为20190301。这是因为第一个事件发生在时间戳=20190101,并且在0和1成为朋友之后,我们有友谊群组[0,1],[2],[3],[4],[5]。然后第二个事件发生在时间戳=20190104,并且在3和4成为朋友之后,我们有以下友谊群组[0,1],[2],[3,4],[5]。之后第三个事件发生在时间戳=20190107,并且在2和3成为朋友之后,我们有以下友谊群组[0,1],[2,3,4],[5]。然后第四个事件发生在时间戳=20190211,并且在1和5成为朋友之后,我们有以下友谊群组[0,1,5],[2,3,4]。之后第五个事件发生在时间戳=20190224,由于2和4已经是朋友,所以任何事情都不会发生。最后,第六个事件发生在时间戳=20190301,并且在0和3成为朋友之后,所有人就都成为朋友了。

为了解决这个问题,我们将遵循以下步骤:

  • 定义一个名为find()的方法,它将接收一个值x,其工作原理如下:

  • 如果parents[x]为-1,则返回x

  • parents[x] := find(parents[x])

  • 返回parents[x]

  • 在主方法中,其工作原理如下:

  • 定义两个名为parents和rank的数组,大小为N,用-1填充parents,用1填充rank

  • 对日志进行排序

  • 对于日志中的每个元素i:

    • 对i[1]和i[2]执行合并操作

    • 查找find(i[2])和find(i[1])

    • 如果rank数组中存在N,则返回i[0]

  • 返回-1

示例 (C++)

让我们看看下面的实现,以便更好地理解:

 在线演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int earliestAcq(vector<vector<int>>& logs, int N) {
      vector<int> ds (N, -1);
      sort(begin(logs), end(logs));
      for(vector<int> &k : logs) {
         if(un(k[1], k[2], ds) == N) return k[0];
      }
      return -1;
   }
   int un(int u, int v, vector<int> & ds) {
      u = find(u, ds);
      v = find(v, ds);
      if(u != v) {
         ds[v] += ds[u];
         ds[u] = v;
      }
      return -ds[v];
   }
   int find(int u, vector<int> & ds) {
      return ds[u] < 0? u : ds[u] = find(ds[u], ds);
   }
};
main(){
   vector<vector<int>> v = {
      {20190101,0,1},{20190104,3,4},{20190107,2,3},{20190211,1,5},
      {20190224,2,4},{20190301,0,3},{20190312,1,2},{20190322,4,5}
   };
   Solution ob;
   cout <<ob.earliestAcq(v, 6);
}

输入

[[20190101,0,1],[20190104,3,4],[20190107,2,3],[20190211,1,5],
[20190224,2,4],[20190301,0,3],[20190312,1,2],[20190322,4,5]]
6

输出

20190301

更新于:2020年4月30日

94 次浏览

开启您的职业生涯

完成课程后获得认证

开始
广告