题目
列表 arr 由在范围 [1, n] 中的所有整数组成,并按严格递增排序。请你对 arr 应用下述算法:
从左到右,删除第一个数字,然后每隔一个数字删除一个,直到到达列表末尾。 重复上面的步骤,但这次是从右到左。也就是,删除最右侧的数字,然后剩下的数字每隔一个删除一个。 不断重复这两步,从左到右和从右到左交替进行,直到只剩下一个数字。 给你整数 n ,返回 arr 最后剩下的数字。
示例:
输入:n = 9
输出:6
解释:
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
arr = [2, 4, 6, 8]
arr = [2, 6]
arr = [6]
来源:力扣(LeetCode)
解答
直接用模拟方法会超时,而这道题归根结底是一道数学问题,数学规律归纳如下:
- 给定大小为
n
的数字构成的1-n
的数组,在进行一轮删除后,剩下的个数为 n / 2 (向下取整); - 剩下的数组仍然是等差数组,且每次公差翻倍,即从1, 2, 4, ...
- 剩下的数组中,要么是原数组第一个a0,要么是原数组第二个a1(即a0+公差):
具体第一个元素规律如下:
- 从右往左进行一轮删除,且个数为偶数个:剩下第一个元素a0
- 其他(从左往右,无论是奇数还是偶数;或从右往左奇数个):剩下第一个元素均为a0+公差。
具体代码如下:
class Solution {
public:
int lastRemaining(int n) {
int a = 1;
int d = 1;
int num = n;
int dir = 1; // 1 从左往右, -1 从右往左
while (num != 1) {
if (!(dir == -1 && num % 2 == 0)) {
a = a + d;
}
dir *= -1;
d *= 2;
num /= 2;
}
return a;
}
};