LeetCode 390. Elimination Game

原题链接在这里:https://leetcode.com/problems/elimination-game/

题目:

There is a list of sorted integers from 1 to n. Starting from left to right, remove the first number and every other number afterward until you reach the end of the list.

Repeat the previous step again, but this time from right to left, remove the right most number and every other number from the remaining numbers.

We keep repeating the steps again, alternating left to right and right to left, until a single number remains.

Find the last number that remains starting with a list of length n.

Example:

Input:
n = 9,
1 2 3 4 5 6 7 8 9
2 4 6 8
2 6
6

Output:
6

题解:

Only record the head, while n > 1 keep loop.

When n == 1, return head.

We need to move head in 2 conditions. One is now it is left to right, the other is now it is right to left and remaining n is odd.

We need to move head by step. Each time step *= 2 as n /= 2.

Time Complexity: O(logn).

Space: O(1).

AC Java:

class Solution {
    public int lastRemaining(int n) {
        boolean lr = true;
        int head = 1;
        int step = 1;
        while(n > 1){
            if(lr || n % 2 == 1){
                head += step;
            }
            
            n /= 2;
            step *= 2;
            lr = !lr;
        }
        
        return head;
    }
}

相关推荐