ZBLOG

class Solution { public: vector visited; std::vector optionalList(std::string &s, const vector & deadends) { vector result; // -1 for (int i = 0; i < 4; i++) { s...

这段代码中可能会导致超时的原因是,在每一次循环中,都将已经访问过的字符串保存在visited数组中,并且使用std::find()函数进行查找。随着循环的进行,visited数组会变得越来越大,而每次查找操作的时间复杂度是O(n),其中n是visited数组的大小。

为了优化这个问题,可以考虑使用哈希表(unordered_set)来存储已经访问过的字符串。哈希表的插入、删除和查找操作的平均时间复杂度是O(1)。修改代码如下:

  1. class Solution {
  2. public:
  3. std::unordered_set<std::string> visited;
  4. std::vector<std::string> optionalList(const std::string &s, const std::vector<std::string> &deadends) {
  5. std::vector<std::string> result;
  6. for (int i = 0; i < 4; i++) {
  7. char c = s[i];
  8. // -1
  9. s[i] = (c == '0' ? '9' : c - 1);
  10. if (visited.find(s) == visited.end() && std::find(deadends.begin(), deadends.end(), s) == deadends.end()) {
  11. result.push_back(s);
  12. visited.insert(s);
  13. }
  14. // +1
  15. s[i] = (c == '9' ? '0' : c + 1);
  16. if (visited.find(s) == visited.end() && std::find(deadends.begin(), deadends.end(), s) == deadends.end()) {
  17. result.push_back(s);
  18. visited.insert(s);
  19. }
  20. // revert s[i] back to the original value
  21. s[i] = c;
  22. }
  23. return result;
  24. }
  25. int openLock(std::vector<std::string>& deadends, const std::string& target) {
  26. std::string init = "0000";
  27. if (std::find(deadends.begin(), deadends.end(), init) != deadends.end()) {
  28. return -1; // the initial state is a deadend
  29. }
  30. visited.insert(init);
  31. std::queue<std::string> q;
  32. q.push(init);
  33. int depth = 0;
  34. while (!q.empty()) {
  35. int k = q.size();
  36. while(k--) {
  37. const std::string p = q.front();
  38. q.pop();
  39. if (p == target) {
  40. return depth;
  41. }
  42. std::vector<std::string> optionalList = optionalList(p, deadends);
  43. for (const auto& item : optionalList) {
  44. visited.insert(item);
  45. q.push(item);
  46. }
  47. }
  48. depth++;
  49. }
  50. return -1;
  51. }
  52. };

这样修改之后,对已经访问过的字符串的查找操作的时间复杂度将降为O(1),可以避免超时问题。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=9919

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?