【LeetCode周赛】第192场周赛

一、1470. 重新排列数组

给你一个数组 nums ,数组中有 2n 个元素,按 [x1,x2,...,xn,y1,y2,...,yn] 的格式排列。

请你将数组按 [x1,y1,x2,y2,...,xn,yn] 格式重新排列,返回重排后的数组。

示例:

输入:nums = [2,5,1,3,4,7], n = 3
输出:[2,3,5,4,1,7]
解释:由于 x1=2, x2=5, x3=1, y1=3, y2=4, y3=7 ,所以答案为 [2,3,5,4,1,7]

输入:nums = [1,2,3,4,4,3,2,1], n = 4
输出:[1,4,2,3,3,2,4,1]

输入:nums = [1,1,2,2], n = 2
输出:[1,2,1,2]

分析:

​ 相当于将数组的奇数和偶数位元素分开,依次插入。没有想到可以原地修改的办法。

代码(Python):

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        ans = [0] * (2 * n)
        even, odd = 0, 1
        i = 0
        j = 0
        while odd < 2 * n:
            ans[odd] = nums[n + i]
            odd += 2
            i += 1
        while even < 2 * n - 1:
            ans[even] = nums[j]
            even += 2
            j += 1
        return ans

二、1471. 数组中的 k 个最强值

给你一个整数数组 arr 和一个整数 k 。

设 m 为数组的中位数,只要满足下述两个前提之一,就可以判定 arr[i] 的值比 arr[j] 的值更强:

  • | arr[i] - m | > | arr[j] - m |

  • |arr[i] - m| == |arr[j] - m|,且 arr[i] > arr[j]
    请返回由数组中最强的 k 个值组成的列表。答案可以以 任意顺序 返回。

中位数 是一个有序整数列表中处于中间位置的值。形式上,如果列表的长度为 n ,那么中位数就是该有序列表(下标从 0 开始)中位于 ((n - 1) / 2) 的元素。

  • 例如 arr = [6, -3, 7, 2, 11],n = 5:数组排序后得到 arr = [-3, 2, 6, 7, 11] ,数组的中间位置为 m = ((5 - 1) / 2) = 2 ,中位数 arr[m] 的值为 6 。

  • 例如 arr = [-7, 22, 17, 3],n = 4:数组排序后得到 arr = [-7, 3, 17, 22] ,数组的中间位置为 m = ((4 - 1) / 2) = 1 ,中位数 arr[m] 的值为 3 。

示例:

输入:arr = [1,2,3,4,5], k = 2
输出:[5,1]
解释:中位数为 3,按从强到弱顺序排序后,数组变为 [5,1,4,2,3]。最强的两个元素是 [5, 1]。[1, 5] 也是正确答案。
注意,尽管 |5 - 3| == |1 - 3| ,但是 5 比 1 更强,因为 5 > 1 。

输入:arr = [1,1,3,5,5], k = 2
输出:[5,5]
解释:中位数为 3, 按从强到弱顺序排序后,数组变为 [5,5,1,1,3]。最强的两个元素是 [5, 5]。

分析:

​ 需要排序来完成,可以利用语言中自带的高级排序功能。

代码(Python):

from typing import List
from functools import cmp_to_key


class Solution:
    def getStrongest(self, arr: List[int], k: int) -> List[int]:
        arr.sort()
        mid = arr[(len(arr) - 1) // 2]

        def strong_sort(x, y):
            if abs(x - mid) == abs(y - mid):
                return y - x
            return abs(y - mid) - abs(x - mid)

        arr.sort(key=cmp_to_key(strong_sort))
        return arr[:k]
from typing import List


class Solution:
    def getStrongest(self, arr: List[int], k: int) -> List[int]:
        arr.sort()
        mid = arr[(len(arr) - 1) // 2]

        def strong_sort(x):
            return (-abs(x - mid), -x)

        arr = sorted(arr,key=strong_sort)
        return arr[:k]

注:Python3中sort函数已经不再支持cmp(compare)参数。如果需要使用,可以引入functools包中的cmp_to_key。或者可以使用代码2中的key,写法少有不同。


三、1472. 设计浏览器历史记录

你有一个只支持单个标签页的 浏览器 ,最开始你浏览的网页是 homepage ,你可以访问其他的网站 url ,也可以在浏览历史中后退 steps 步或前进 steps 步。

请你实现 BrowserHistory 类:

  • BrowserHistory(string homepage) ,用 homepage 初始化浏览器类。

  • void visit(string url) 从当前页跳转访问 url 对应的页面 。执行此操作会把浏览历史前进的记录全部删除。

  • string back(int steps) 在浏览历史中后退 steps 步。如果你只能在浏览历史中后退至多 x 步且 steps > x ,那么你只后退 x 步。请返回后退 至多 steps 步以后的 url 。

  • string forward(int steps) 在浏览历史中前进 steps 步。如果你只能在浏览历史中前进至多 x 步且 steps > x ,那么你只前进 x 步。请返回前进 至多 steps步以后的 url 。

示例:

输入:
["BrowserHistory","visit","visit","visit","back","back","forward","visit","forward","back","back"]
[["leetcode.com"],["google.com"],["facebook.com"],["youtube.com"],[1],[1],[1],["linkedin.com"],[2],[2],[7]]
输出:
[null,null,null,null,"facebook.com","google.com","facebook.com",null,"linkedin.com","google.com","leetcode.com"]

解释:
BrowserHistory browserHistory = new BrowserHistory("leetcode.com");
browserHistory.visit("google.com"); // 你原本在浏览 "leetcode.com" 。访问 "google.com"
browserHistory.visit("facebook.com"); // 你原本在浏览 "google.com" 。访问 "facebook.com"
browserHistory.visit("youtube.com"); // 你原本在浏览 "facebook.com" 。访问 "youtube.com"
browserHistory.back(1); // 你原本在浏览 "youtube.com" ,后退到 "facebook.com" 并返回 "facebook.com"
browserHistory.back(1); // 你原本在浏览 "facebook.com" ,后退到 "google.com" 并返回 "google.com"
browserHistory.forward(1); // 你原本在浏览 "google.com" ,前进到 "facebook.com" 并返回 "facebook.com"
browserHistory.visit("linkedin.com"); // 你原本在浏览 "facebook.com" 。 访问 "linkedin.com"
browserHistory.forward(2); // 你原本在浏览 "linkedin.com" ,你无法前进任何步数。
browserHistory.back(2); // 你原本在浏览 "linkedin.com" ,后退两步依次先到 "facebook.com" ,然后到 "google.com" ,并返回 "google.com"
browserHistory.back(7); // 你原本在浏览 "google.com", 你只能后退一步到 "leetcode.com" ,并返回 "leetcode.com"

分析:

​ 当时这道题没有做出来。有一个关键的地方没有理解到位。

例如示例中:当页面处在 facebook时访问了 linkedin,这时就无法向前到达原来的 youtube

​ 所以不需要复杂的数据结构。列表就可以很好的解决这个问题。

代码(Python):

class BrowserHistory:

    def __init__(self, homepage: str):
        self.history = [homepage]
        self.index = 0
        self.last = self.index + 1

    def visit(self, url: str) -> None:
        # 如果超出列表,就使用append扩展一下
        if self.index == len(self.history) - 1:
            self.history.append(url)
        else:
            self.history[self.index + 1] = url
        self.index += 1
        self.last = self.index + 1

    def back(self, steps: int) -> str:
        steps = min(self.index, steps)
        self.index -= steps
        return self.history[self.index]

    def forward(self, steps: int) -> str:
        # last储存列表长度,这一步是为了查看steps是否溢出
        steps = min(self.last - self.index - 1, steps)
        self.index += steps
        return self.history[self.index]

四、1473. 给房子涂色 III:

在一个小城市里,有 m 个房子排成一排,你需要给每个房子涂上 n 种颜色之一(颜色编号为 1 到 n )。有的房子去年夏天已经涂过颜色了,所以这些房子不需要被重新涂色。

我们将连续相同颜色尽可能多的房子称为一个街区。(比方说 houses = [1,2,2,3,3,2,1,1] ,它包含 5 个街区 [{1}, {2,2}, {3,3}, {2}, {1,1}] 。)

给你一个数组 houses ,一个 m * n 的矩阵 cost 和一个整数 target ,其中:

houses[i]:是第 i 个房子的颜色,0 表示这个房子还没有被涂色。
cost[i][j]:是将第 i 个房子涂成颜色 j+1 的花费。
请你返回房子涂色方案的最小总花费,使得每个房子都被涂色后,恰好组成 target 个街区。如果没有可用的涂色方案,请返回 -1 。

示例:

输入:houses = [0,0,0,0,0], cost = [[1,10],[10,1],[10,1],[1,10],[5,1]], m = 5, n = 2, target = 3
输出:9
解释:房子涂色方案为 [1,2,2,1,1]
此方案包含 target = 3 个街区,分别是 [{1}, {2,2}, {1,1}]。
涂色的总花费为 (1 + 1 + 1 + 1 + 5) = 9。

分析:

​ 这道题是一道很明显的动态规划,但是需要一个高维的dp数组。能力有限,比赛后看着大佬的题解写的Golang版本。状态转移方程和一些关键步骤都在注释中写了一些,以后理解加深了再来补。

代码(Golang):

func minCost(houses []int, cost [][]int, m int, n int, target int) int {
	// 三维dp数组初始化
	// dp[i][j][k]表示第i个房子涂第j个颜色,前i个房子形成了k个社区
	dp := make([][][]int, m)
	for i := 0; i < m; i++ {
		dp[i] = make([][]int, n + 1)
		for j := 0; j < n + 1; j++ {
			dp[i][j] = make([]int, target + 1)
			for k := 0; k < target + 1; k++ {
				dp[i][j][k] = math.MaxInt32
			}
		}
	}

	//// 如果房子未涂色,需要重新涂色
	if houses[0] == 0 {
		for i := 1; i < n + 1; i++ {
			// 颜色可以随意涂
			dp[0][i][1] = cost[0][i - 1]
		}
	}

	// dp状态转移
	// k的值是由当前房子颜色和前一个房子颜色作比较得到的:
	// 1.i号房子颜色和i-1号房子颜色相同,i,k的值从i-1,k得到
	// 2.i号房子颜色和i-1号房子颜色不同,i,k的值从i-1,k-1得到
	for i := 1; i < m; i++ {
		if houses[i] == 0 {
			for j1 := 1; j1 < n + 1; j1++ {
				for k := 1; k < target + 1; k++ {
					for j2 := 1; j2 < n + 1; j2++ {
						if j1 == j2 {
							dp[i][j1][k] = min(dp[i][j1][k], dp[i - 1][j2][k] + cost[i][j1 - 1])
						} else {
							dp[i][j1][k] = min(dp[i][j1][k], dp[i - 1][j2][k - 1] + cost[i][j1 -1])
						}
					}
				}
			}
		} else {  // 不需要涂色的也需要更新dp数组
			for k := 1; k < target + 1; k++ {
				for j2 := 1; j2 < n + 1; j2++ {
					j1 := houses[i]
					if j1 == j2 {
						dp[i][j1][k] = min(dp[i][j1][k], dp[i - 1][j2][k])
					} else {
						dp[i][j1][k] = min(dp[i][j1][k], dp[i - 1][j2][k - 1])
					}
				}
			}
		}
	}
	ans := math.MaxInt32
	// 寻找满足社区数且涂色费用最低的涂色方案
	for i := 1; i < n+ 1; i++ {
		ans = min(ans, dp[m - 1][i][target])
	}
	if ans == math.MaxInt32 {
		ans = -1
	}
	return ans
}

func min(nums ...int) int {
	min := nums[0]
	for i := 1; i < len(nums); i++ {
		if nums[i] < min {
			min = nums[i]
		}
	}
	return min
}

posted @ 2020-06-12 00:11  macguz  阅读(171)  评论(0编辑  收藏  举报