小明陪小红去看钻石,他们从一堆钻石中随机抽取两颗并比较她们的重量。这些钻石的重量各不相同。在他们们比较了一段时间后,它们看中了两颗钻石g1和g2。现在请你根据之前比较的信息判断这两颗钻石的哪颗更重。
给定两颗钻石的编号g1,g2,编号从1开始,同时给定关系数组vector,其中元素为一些二元组,第一个元素为一次比较中较重的钻石的编号,第二个元素为较轻的钻石的编号。最后给定之前的比较次数n。请返回这两颗钻石的关系,若g1更重返回1,g2更重返回-1,无法判断返回0。输入数据保证合法,不会有矛盾情况出现。
测试样例:
2,3,[[1,2],[2,4],[1,3],[4,3]],4
返回: 1
第一次运行时,显示所用内存使用超过限制。后来找到原因为: check函数中没有设置标志,使原来已经push进queue的数又不断的重新push浪费了内存(虽然不会出现死循环,因为小的数不可能再比大的数大,但是开销大大增加)。
#include<unordered_map> class Cmp { public:
// 根据邻接数组判断g2是不是在g1的后面 bool check(int g1, int g2, unordered_map<int, vector<int>>tu){ cout << "**********************************" << endl; if (tu[g1].empty())return false; queue<int> in; map<int, bool> monitor; in.push(g1); while (!in.empty()){ int q = in.front(); in.pop(); monitor[q] = true; cout << "q is:" << q << endl; if (q == g2) return true; for (int i = 0; i < tu[q].size(); i++){ if (tu[q][i] == g2) return true; if (!monitor[tu[q][i]]) in.push(tu[q][i]); } } return false; }
// cmp 函数,将records用邻接数组表示 int cmp(int g1, int g2, vector<vector<int> > records, int n) { // write code here unordered_map<int, vector<int>> tu; for (int i = 0; i < records.size(); i++){ tu[records[i][0]].push_back(records[i][1]); } if (check(g1, g2, tu)) return 1; else if (check(g2, g1, tu)) return -1; else return 0; } };