数轴上放置了一些筹码,每个筹码的位置存在数组 chips
当中。
你可以对 任何筹码 执行下面两种操作之一(不限操作次数,0 次也可以):
- 将第
i
个筹码向左或者右移动 2 个单位,代价为 0。 - 将第
i
个筹码向左或者右移动 1 个单位,代价为 1。
最开始的时候,同一位置上也可能放着两个或者更多的筹码。
返回将所有筹码移动到同一位置(任意位置)上所需要的最小代价。
示例 1:
输入:chips = [1,2,3]
输出:1
解释:第二个筹码移动到位置三的代价是 1,第一个筹码移动到位置三的代价是 0,总代价为 1。
示例 2:
输入:chips = [2,2,2,3,3]
输出:2
解释:第四和第五个筹码移动到位置二的代价都是 1,所以最小总代价为 2。
提示:
1 <= chips.length <= 100
1 <= chips[i] <= 10^9
解题思路
首先要思考这个最小代价怎么出现?一定是通过一些操作(并且这些操作的代价最小)使得最后的筹码间隔奇数位置堆叠(这种做法的移动代价是0
)。那么这就必然会出现两种结果:一种是最后所有筹码出现在奇数位置,二是最后所有筹码出现在偶数位置。
那么我们只需要统计一遍chips
,然后判断奇数位置筹码的和和偶数位置筹码的和哪个小即可。
class Solution:
def minCostToMoveChips(self, chips: List[int]) -> int:
cnt = collections.Counter(chips)
r1, r2 = 0, 0
for k, v in cnt.items():
if k & 1:
r1 += v
else:
r2 += v
return min(r1, r2)
当然可以写得更简洁
class Solution:
def minCostToMoveChips(self, chips: List[int]) -> int:
res = [0, 0]
for c in chips:
res[c % 2] += 1
return min(res)
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!