Using ternary operator

amarchourasia98 · amarchourasia98 · commit 533dc1c7923b · 2020-10-01T23:19:12.000+05:30
diff --git a/src/main/java/com/leetcode/arrays/MajorityElement.java b/src/main/java/com/leetcode/arrays/MajorityElement.java
@@ -20,23 +20,19 @@ public class MajorityElement {
      * @param nums
      * @return
      */
-    public static int majorityElement(int[] nums) {
-        int count = 1;
-        int majElem = nums[0];
+    public int majorityElement(int[] nums) {
+        int count = 0;
+        Integer candidate = null;
 
-        for (int i = 1; i < nums.length; i++) {
-            if (count <= 0) {
-                majElem = nums[i];
-                count = 0;
-            }
-            if (majElem == nums[i]) {
-                count++;
-            } else {
-                count--;
+        for (int num : nums) {
+            if (count == 0) {
+                candidate = num;
             }
+            count += (num == candidate) ? 1 : -1;
         }
 
-        return majElem;
+        return candidate;
+       
     }
 
     public static void main(String[] args) {
