LeetCode/E1046_LastStoneWeight.java at master · Abhilash-Mandlekar/LeetCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
class Solution {
    public int lastStoneWeight(int[] stones) {

        List<Integer> st = new ArrayList<>();
        int temp;

        for(int i=0; i<stones.length; i++){
            st.add(stones[i]);
        }

        while(st.size()>1){
            Collections.sort(st);
            //System.out.println(st);
            if(st.get(st.size()-2)<st.get(st.size()-1)){
                temp = st.get(st.size()-1) - st.get(st.size()-2);
                st.remove(st.size()-1); st.remove(st.size()-1);
                st.add(temp);
            }
            else if (st.get(st.size()-2)>st.get(st.size()-1)){
                temp = st.get(st.size()-2) - st.get(st.size()-1);
                st.remove(st.size()-1); st.remove(st.size()-1);
                st.add(temp);
            }
            else{
                st.remove(st.size()-1); st.remove(st.size()-1);
            }
        }

        if(st.size()==0)
            return 0;
        else
            return st.get(0);
    }

}