跪求大佬帮我看看代码: LeetCode706 Design HashMap)

本人Java菜niao一枚，最近刚开始用Java刷题。刚学了一下MIT6.006的Hash Map的部分，就来练了一下力扣的706 （https://leetcode.com/problems/design-hashmap/）练练手。结果submit以后发现有wrong answer。苦思冥想了一整天还是没搞清楚自己错在哪里，跪求地里的高手指点一二。有指点必加米。先谢谢了！

1. 
   
2. class ListNode {
   
3.     public int key;
   
4.     public int value;
   
5.     public ListNode next;
   
6.    
   
7.     public ListNode(int key, int value) {
   
8.         this.key = key;
   
9.         this.value = value;
   
10.     }
    
11. }
    
12. 
    
13. class MyHashMap {
    
14.     private ListNode[] arr;
    
15.     private int divisor = 2069;
    
16. 
    
17.     /** Initialize your data structure here. */
    
18.     public MyHashMap() {
    
19.         this.arr = new ListNode[divisor];
    
20.     }
    
21.    
    
22.     /** value will always be non-negative. */
    
23.     public void put(int key, int value) {
    
24.         int k = key%this.divisor;
    
25.         if (arr[k]==null){
    
26.             arr[k] = new ListNode(key, value);
    
27.         } else {
    
28.             ListNode cur = arr[k], prev = null;
    
29.             while (cur!=null){
    
30.                 if (cur.key == key) {
    
31.                     cur.value = value;
    
32.                     return;
    
33.                 }
    
34.                 prev = cur;
    
35.                 cur = cur.next;
    
36.             }
    
37.             prev.next = new ListNode(key, value);
    
38.         }
    
39.     }
    
40.    
    
41.     /** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */
    
42.     public int get(int key) {
    
43.         int k = key%this.divisor;
    
44.         if (arr[k]==null) {
    
45.             return -1;
    
46.         } else {
    
47.             ListNode cur = arr[k];
    
48.             while (cur!=null) {
    
49.                 if (cur.key == key) {
    
50.                     return cur.value;
    
51.                 }
    
52.                 cur = cur.next;
    
53.             }
    
54.         }
    
55.         
    
56.         return -1;
    
57.     }
    
58.    
    
59.     /** Removes the mapping of the specified value key if this map contains a mapping for the key */
    
60.     public void remove(int key) {
    
61.         int k = key%this.divisor;
    
62.         if (arr[k]!=null) {
    
63.             ListNode cur = arr[k], prev = null;
    
64.             while (cur!=null) {
    
65.                 if (cur.key == key) {
    
66.                     if (prev!=null) {
    
67.                         prev.next = cur.next;
    
68.                     } else {
    
69.                         arr[k] = null;
    
70.                     }
    
71.                     cur = null;
    
72.                     break;
    
73.                 }
    
74.                 prev = cur;
    
75.                 cur = cur.next;
    
76.             }
    
77.         }
    
78.     }
    
79. }
    
80. 
    
81. /**
    
82. * Your MyHashMap object will be instantiated and called as such:
    
83. * MyHashMap obj = new MyHashMap();
    
84. * obj.put(key,value);
    
85. * int param_2 = obj.get(key);
    
86. * obj.remove(key);
    
87. */

复制代码

BobbyBear 发表于 2020-7-27 22:22
对啊！你这样一提醒我就突然看到问题了。多谢大佬！

没事，我也不是大佬，就是逛论坛走过顺便交流一下XD

楼主，我觉得如果你是删除链表的第一个节点，应该把后面的节点接到arr[k]吧？while loop 里面的else改一下这样试试？

1. public void remove(int key) {
   
2.         int k = key%this.divisor;
   
3.         if (arr[k]!=null) {
   
4.             ListNode cur = arr[k], prev = null;
   
5.             while (cur!=null) {
   
6.                 if (cur.key == key) {
   
7.                     if (prev!=null) {
   
8.                         prev.next = cur.next;
   
9.                     } else {
   
10.                         ListNode nextNode = cur.next;
    
11.                         arr[k] = nextNode ;
    
12.                     }
    
13.                     cur = null;
    
14.                     break;
    
15.                 }
    
16.                 prev = cur;
    
17.                 cur = cur.next;
    
18.             }
    
19.         }
    
20.     }

复制代码

hjy 发表于 2020-7-27 13:31
楼主，我觉得如果你是删除链表的第一个节点，应该把后面的节点接到arr[k]吧？while loop 里面的else改一下 ...

对啊！你这样一提醒我就突然看到问题了。多谢大佬！