LeetCode 1686. 石子游戏 VI【排序,贪心】【Py3,Go】2000

2024-02-03 09:12

本文主要是介绍LeetCode 1686. 石子游戏 VI【排序,贪心】【Py3,Go】2000,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

本文属于「征服LeetCode」系列文章之一,这一系列正式开始于2021/08/12。由于LeetCode上部分题目有锁,本系列将至少持续到刷完所有无锁题之日为止;由于LeetCode还在不断地创建新题,本系列的终止日期可能是永远。在这一系列刷题文章中,我不仅会讲解多种解题思路及其优化,还会用多种编程语言实现题解,涉及到通用解法时更将归纳总结出相应的算法模板。

为了方便在PC上运行调试、分享代码文件,我还建立了相关的仓库:https://github.com/memcpy0/LeetCode-Conquest。在这一仓库中,你不仅可以看到LeetCode原题链接、题解代码、题解文章链接、同类题目归纳、通用解法总结等,还可以看到原题出现频率和相关企业等重要信息。如果有其他优选题解,还可以一同分享给他人。

由于本系列文章的内容随时可能发生更新变动,欢迎关注和收藏征服LeetCode系列文章目录一文以作备忘。

Alice 和 Bob 轮流玩一个游戏,Alice 先手。

一堆石子里总共有 n 个石子,轮到某个玩家时,他可以 移出 一个石子并得到这个石子的价值。Alice 和 Bob 对石子价值有 不一样的的评判标准 。双方都知道对方的评判标准。

给你两个长度为 n 的整数数组 aliceValuesbobValuesaliceValues[i]bobValues[i] 分别表示 Alice 和 Bob 认为第 i 个石子的价值。

所有石子都被取完后,得分较高的人为胜者。如果两个玩家得分相同,那么为平局。两位玩家都会采用 最优策略 进行游戏。

请你推断游戏的结果,用如下的方式表示:

  • 如果 Alice 赢,返回 1
  • 如果 Bob 赢,返回 -1
  • 如果游戏平局,返回 0

示例 1:

输入:aliceValues = [1,3], bobValues = [2,1]
输出:1
解释:
如果 Alice 拿石子 1 (下标从 0开始),那么 Alice 可以得到 3 分。
Bob 只能选择石子 0 ,得到 2 分。
Alice 获胜。

示例 2:

输入:aliceValues = [1,2], bobValues = [3,1]
输出:0
解释:
Alice 拿石子 0 , Bob 拿石子 1 ,他们得分都为 1 分。
打平。

示例 3:

输入:aliceValues = [2,4,3], bobValues = [1,6,7]
输出:-1
解释:
不管 Alice 怎么操作,Bob 都可以得到比 Alice 更高的得分。
比方说,Alice 拿石子 1 ,Bob 拿石子 2 , Alice 拿石子 0 ,Alice 会得到 6 分而 Bob 得分为 7 分。
Bob 会获胜。

提示:

  • n == aliceValues.length == bobValues.length
  • 1 <= n <= 10^5
  • 1 <= aliceValues[i], bobValues[i] <= 100

解法 排序+贪心

设 Alice 选的数字之和为 A A A,Bob 选的数字之和为 B B B

  • 如果 A − B > 0 A−B>0 AB>0 那么 Alice 赢;
  • 如果 A − B < 0 A−B<0 AB<0 那么 Bob 赢;
  • 如果 A − B = 0 A-B=0 AB=0 则平局。
  • 所以 Alice 需要最大化 A − B A−B AB ,Bob 需要最小化 A − B A-B AB

下文把数组 aliceValues \textit{aliceValues} aliceValues 记作 A A A,把数组 bobValues \textit{bobValues} bobValues 记作 B B B

a = [ 2 , 4 , 3 , 5 ] , b = [ 1 , 6 , 7 , 1 ] a=[2,4,3,5], b=[1,6,7,1] a=[2,4,3,5],b=[1,6,7,1] 为例说明。

假设 Bob 把所有石子都拿走,则 A = 0 , B = 15 , A − B = − 15 A=0,\ B=15,\ A−B=−15 A=0, B=15, AB=15

先来想一想,如果 Alice 只能拿走一颗石子,应该拿走哪颗呢?

  1. 拿走第一颗,那么 A = 2 , B = 14 , A − B = − 12 A=2,\ B=14,\ A-B=-12 A=2, B=14, AB=12
  2. 拿走第二颗,那么 A = 4 , B = 9 , A − B = − 5 A=4,\ B=9,\ A-B=-5 A=4, B=9, AB=5
  3. 拿走第三颗,那么 A = 3 , B = 8 , A − B = − 5 A=3,\ B=8,\ A-B=-5 A=3, B=8, AB=5
  4. 拿走第四颗,那么 A = 5 , B = 14 , A − B = − 9 A=5,\ B=14,\ A-B=-9 A=5, B=14, AB=9

对比 Bob 把所有石子都拿走的情况,如果 Alice 拿走第二颗或者第三颗,都可以让 − 15 -15 15 增大为 − 5 -5 5 ,增量为 10 10 10 。由于 A A A 增加了 a [ i ] a[i] a[i] B B B 减少了 b [ i ] b[i] b[i] ,所以 A − B A−B AB 的增量等于
a [ i ] − ( − b [ i ] ) = a [ i ] + b [ i ] a[i] - (-b[i]) = a[i] + b[i] a[i](b[i])=a[i]+b[i]
所以 Alice 拿走 a [ i ] + b [ i ] a[i]+b[i] a[i]+b[i] 最大的石子最优。

回到原题,Alice 可以拿走两颗石子,应该拿走哪两颗呢?

我们从 Bob 把所有石子都拿走的情况出发,也就是在 A = 0 , B = 15 A=0,\ B=15 A=0, B=15 的基础上思考,Alice 拿走哪两颗石子,可以让 A − B A-B AB 增加的尽量多?

定义 c [ i ] = a [ i ] + b [ i ] c[i]=a[i]+b[i] c[i]=a[i]+b[i] ,那么 c = [ 3 , 10 , 10 , 6 ] c=[3,10,10,6] c=[3,10,10,6] 。现在问题变成:给定数组 c c c ,Alice 每回合拿走一个数,Bob 每回合删除一个数,Alice 拿走的数之和最大是多少?注意 Bob 要让 Alice 拿走的数之和尽量小。

如此转换后,贪心策略就很显然了:Alice 从大到小拿走数字,Bob 也从大到小删除数字。

所以把 c c c 从大到小排序为 [ 10 , 10 , 6 , 3 ] [10,10,6,3] [10,10,6,3] ,两人从左往右交替取数,那么 Alice 只能拿走下标为偶数的数字,其余数字被 Bob 删除。所以 A − B A-B AB 最大可以增加 c [ 0 ] + c [ 2 ] = 10 + 6 = 16 c[0]+c[2]=10+6=16 c[0]+c[2]=10+6=16增加后 A − B = 1 > 0 A-B=1>0 AB=1>0 ,Alice 险胜!

算法
把数组按照 a [ i ] + b [ i ] a[i]+b[i] a[i]+b[i] 从大到小排序。可以创建一个 ( a [ i ] , b [ i ] ) (a[i],b[i]) (a[i],b[i])pair 数组对其排序,也可以创建一个下标数组排序。

diff \textit{diff} diff 表示 A − B A-B AB ,初始化 diff = 0 \textit{diff}=0 diff=0 。遍历数组,把偶数下标的 a [ i ] a[i] a[i] 加到 A A A 中,相当于 d i f f diff diff 增加了 a [ i ] a[i] a[i] ;把奇数下标的 b [ i ] b[i] b[i] 加到 B B B 中,相当于 diff \textit{diff} diff 减少了 b [ i ] b[i] b[i]

循环结束后,如果 d i f f > 0 diff>0 diff>0 ,返回 1 1 1 ;如果 d i f f < 0 diff<0 diff<0 ,返回 − 1 −1 1 ;如果 d i f f = 0 diff=0 diff=0 ,返回 0 0 0

问:从这个思路的本质是什么?为什么这样转换一下,问题就变得简单了许多?
答:转换前,我们需要同时考虑 a [ i ] a[i] a[i] b [ i ] b[i] b[i] 这两个变量,不好处理。转换成 Bob 先把所有数字选了,我们就只需要思考 Alice 如何选数字,只有 c [ i ] c[i] c[i] 这一个变量,更容易处理。从某种程度上来说,这也可以算作一种「正难则反」。

问:有没有其它的思考方式?
答:也可以这样思考:对比两颗石子 ( a [ i ] , b [ i ] ) (a[i],b[i]) (a[i],b[i]) ( a [ j ] , b [ j ] ) (a[j],b[j]) (a[j],b[j]) 。如果 Alice 选 i i i ,Bob 选 j j j ,那么 A − B = a [ i ] − b [ j ] A−B=a[i]−b[j] AB=a[i]b[j] ;如果 Alice 选 j j j ,Bob 选 i i i ,那么 A − B = a [ j ] − b [ i ] A−B=a[j]−b[i] AB=a[j]b[i] 。如果 Alice 选 i i i 更优,则有 a [ i ] − b [ j ] > a [ j ] − b [ i ] a[i]−b[j]>a[j]−b[i] a[i]b[j]>a[j]b[i] ,即 a [ i ] + b [ i ] > a [ j ] + b [ j ] a[i]+b[i]>a[j]+b[j] a[i]+b[i]>a[j]+b[j] ,说明 Alice 应当优先选 a [ i ] + b [ i ] a[i]+b[i] a[i]+b[i] 更大的石子。

# python
class Solution:def stoneGameVI(self, a: List[int], b: List[int]) -> int:pairs = sorted(zip(a, b), key = lambda p: -(p[0] + p[1])) # a[i]+b[i]越大排在前diff = sum(x if i % 2 == 0 else -y for i, (x, y) in enumerate(pairs)) # 表示A-Breturn (diff > 0) - (diff < 0)class Solution:def stoneGameVI(self, a: List[int], b: List[int]) -> int:s = sorted((x + y for x, y in zip(a, b)), reverse = True)diff = sum(s[::2]) - sum(b)return (diff > 0) - (diff < 0)
func stoneGameVI(a []int, b []int) int {type pair struct { x, y int }pairs := make([]pair, len(a)) // pair数组for i, x := range a {pairs[i] = pair{x, b[i]}}slices.SortFunc(pairs, func(p, q pair) int {return q.x + q.y - (p.x + p.y)})diff := 0for i, p := range pairs {if i % 2 == 0 {diff += p.x} else {diff -= p.y}}return cmp.Compare(diff, 0)
}

复杂度分析:

  • 时间复杂度: O ( n log ⁡ n ) \mathcal{O}(n\log n) O(nlogn) ,其中 n n n a a a 的长度。瓶颈在排序上。
  • 空间复杂度: O ( n ) \mathcal{O}(n) O(n)

这篇关于LeetCode 1686. 石子游戏 VI【排序,贪心】【Py3,Go】2000的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/673653

相关文章

go中空接口的具体使用

《go中空接口的具体使用》空接口是一种特殊的接口类型,它不包含任何方法,本文主要介绍了go中空接口的具体使用,具有一定的参考价值,感兴趣的可以了解一下... 目录接口-空接口1. 什么是空接口?2. 如何使用空接口?第一,第二,第三,3. 空接口几个要注意的坑坑1:坑2:坑3:接口-空接口1. 什么是空接

利用Go语言开发文件操作工具轻松处理所有文件

《利用Go语言开发文件操作工具轻松处理所有文件》在后端开发中,文件操作是一个非常常见但又容易出错的场景,本文小编要向大家介绍一个强大的Go语言文件操作工具库,它能帮你轻松处理各种文件操作场景... 目录为什么需要这个工具?核心功能详解1. 文件/目录存javascript在性检查2. 批量创建目录3. 文件

C++快速排序超详细讲解

《C++快速排序超详细讲解》快速排序是一种高效的排序算法,通过分治法将数组划分为两部分,递归排序,直到整个数组有序,通过代码解析和示例,详细解释了快速排序的工作原理和实现过程,需要的朋友可以参考下... 目录一、快速排序原理二、快速排序标准代码三、代码解析四、使用while循环的快速排序1.代码代码1.由快

Go语言中最便捷的http请求包resty的使用详解

《Go语言中最便捷的http请求包resty的使用详解》go语言虽然自身就有net/http包,但是说实话用起来没那么好用,resty包是go语言中一个非常受欢迎的http请求处理包,下面我们一起来学... 目录安装一、一个简单的get二、带查询参数三、设置请求头、body四、设置表单数据五、处理响应六、超

Golang基于内存的键值存储缓存库go-cache

《Golang基于内存的键值存储缓存库go-cache》go-cache是一个内存中的key:valuestore/cache库,适用于单机应用程序,本文主要介绍了Golang基于内存的键值存储缓存库... 目录文档安装方法示例1示例2使用注意点优点缺点go-cache 和 Redis 缓存对比1)功能特性

Go 1.23中Timer无buffer的实现方式详解

《Go1.23中Timer无buffer的实现方式详解》在Go1.23中,Timer的实现通常是通过time包提供的time.Timer类型来实现的,本文主要介绍了Go1.23中Timer无buff... 目录Timer 的基本实现无缓冲区的实现自定义无缓冲 Timer 实现更复杂的 Timer 实现总结在

Go使用pprof进行CPU,内存和阻塞情况分析

《Go使用pprof进行CPU,内存和阻塞情况分析》Go语言提供了强大的pprof工具,用于分析CPU、内存、Goroutine阻塞等性能问题,帮助开发者优化程序,提高运行效率,下面我们就来深入了解下... 目录1. pprof 介绍2. 快速上手:启用 pprof3. CPU Profiling:分析 C

使用Go语言开发一个命令行文件管理工具

《使用Go语言开发一个命令行文件管理工具》这篇文章主要为大家详细介绍了如何使用Go语言开发一款命令行文件管理工具,支持批量重命名,删除,创建,移动文件,需要的小伙伴可以了解下... 目录一、工具功能一览二、核心代码解析1. 主程序结构2. 批量重命名3. 批量删除4. 创建文件/目录5. 批量移动三、如何安

Go路由注册方法详解

《Go路由注册方法详解》Go语言中,http.NewServeMux()和http.HandleFunc()是两种不同的路由注册方式,前者创建独立的ServeMux实例,适合模块化和分层路由,灵活性高... 目录Go路由注册方法1. 路由注册的方式2. 路由器的独立性3. 灵活性4. 启动服务器的方式5.

Go语言中三种容器类型的数据结构详解

《Go语言中三种容器类型的数据结构详解》在Go语言中,有三种主要的容器类型用于存储和操作集合数据:本文主要介绍三者的使用与区别,感兴趣的小伙伴可以跟随小编一起学习一下... 目录基本概念1. 数组(Array)2. 切片(Slice)3. 映射(Map)对比总结注意事项基本概念在 Go 语言中,有三种主要