Update Last Stone Weight.java

varunu28 · web-flow · commit a14f47a2bc1e · 2022-04-07T11:46:46.000+05:30
diff --git a/Easy/Last Stone Weight.java b/Easy/Last Stone Weight.java
@@ -1,16 +1,16 @@
 class Solution {
   public int lastStoneWeight(int[] stones) {
-    PriorityQueue<Integer> pq = new PriorityQueue<>((a, b) -> b - a);
+    PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> o2 - o1);
     for (int stone : stones) {
       pq.add(stone);
     }
     while (pq.size() > 1) {
-      int max1 = pq.poll();
-      int max2 = pq.poll();
-      if (max1 != max2) {
-        pq.add(max1 - max2);
+      int firstStone = pq.poll();
+      int secondStone = pq.poll();
+      if (firstStone != secondStone) {
+        pq.add(firstStone - secondStone);
       }
     }
-    return pq.isEmpty() ? 0 : pq.peek();
+    return pq.isEmpty() ? 0 : pq.poll();
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -1,16 +1,16 @@`
`1`	`1`	`class Solution {`
`2`	`2`	`public int lastStoneWeight(int[] stones) {`
`3`		`- PriorityQueue<Integer> pq = new PriorityQueue<>((a, b) -> b - a);`
	`3`	`+ PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> o2 - o1);`
`4`	`4`	`for (int stone : stones) {`
`5`	`5`	`pq.add(stone);`
`6`	`6`	`}`
`7`	`7`	`while (pq.size() > 1) {`
`8`		`- int max1 = pq.poll();`
`9`		`- int max2 = pq.poll();`
`10`		`- if (max1 != max2) {`
`11`		`- pq.add(max1 - max2);`
	`8`	`+ int firstStone = pq.poll();`
	`9`	`+ int secondStone = pq.poll();`
	`10`	`+ if (firstStone != secondStone) {`
	`11`	`+ pq.add(firstStone - secondStone);`
`12`	`12`	`}`
`13`	`13`	`}`
`14`		`- return pq.isEmpty() ? 0 : pq.peek();`
	`14`	`+ return pq.isEmpty() ? 0 : pq.poll();`
`15`	`15`	`}`
`16`	`16`	`}`