题目描述:
将一个链表通过交换节点位置的操作使链表中的value为k的节点都聚集在一起,给出最少需要的交换次数。
输入描述:
第一行为链表的长度,取值范围为1~1000
第二行为链表的节点,如 1 2 3 代表1->2->3,每个节点value为整数,取值范围0~10000
第三行为k
输出描述:
输出需要的最少交换次数
例子:
输入:
5
1 2 1 2 1
2
输出:
1
解法:
这道题其实链表和数组是一样的,所以直接用数组来存即可。
一开始的思路是,找出这个数组的最长的k元素的重复子串,假设长度为lenght, 然后再用总的k出现的次数减length, 再加1
n = int(input())
array = list(map(int, input().split()))
target = int(input())
nums = 0
for i in range(len(array)):
if array[i] == target:
nums += 1
# 求最长重复字串的长度:
def get(array, target):
# write your code here
n = len(array)
dp = [[0 for j in range(n+1)] for i in range(n+1)]
for i in range(1, n+1):
for j in range(1, n+1):
if array[i-1] == array[j-1] == target and i != j:
dp[i][j] = dp[i-1][j-1] + 1
else:
dp[i][j] = max(dp[i][j-1], dp[i-1][j])
return dp[n][n]
print(nums - get(array, target) + 1)
方法解读:
首先联想 1143. 最长公共子序列,联想一下如何求两个不同字符串序列中的最长公共子序列(对于序列的下标位置没有要求)。就可以看出,这个题目就是求两个相同字符串序列的最长公共子序列,但是有个要求,就是相同字符的下标不能相同。进一步改进的话,就是要等于target。参考——https://www.jianshu.com/p/6581b27f34f8
结果:
最后只AC了9%,思考一下,这样是只考虑了最坏的情况,例如,输入序列是1233333433355 , target是3,实际只需要交换一次就行了。
然后联想到另一题 1151. 最少交换次数来组合所有的 1 , 题目描述为:给出一个二进制数组 data
,你需要通过交换位置,将数组中 任何位置 上的 1 组合到一起,并返回所有可能中所需 最少的交换次数。有一种解法是滑动窗口:
'''
思路和心得:
1.采用固定区间的方法。
假设最后的全1就在当前的窗口
则窗口内0的个数==需要交换的次数
2.套滑动窗口模板,3步走
'''
def minSwaps(data):
n = len(data)
window_len = sum(data) #所有1的个数
cnt1 = sum(data[ :window_len]) #第一个窗口内1的个数
res = window_len - cnt1 #第一个窗口内0的个数
for R in range(window_len, n): #滑动窗口
cnt1 += data[R] #进R
cnt1 -= data[R - window_len]#弹L
cnt0 = window_len - cnt1 #窗口内0的个数==如果以此窗口的位置为最后结果,需要交换的次数
res = min(res, cnt0) #更新res
return res
minSwaps([1,2,1,2,1])
'''
作者:Hanxin_Hanxin
链接:https://leetcode-cn.com/problems/minimum-swaps-to-group-all-1s-together/solution/c-python3-hua-dong-chuang-kou-gu-ding-qu-hohf/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
'''
然后我开始魔改了,把原始输入的array遍历一遍,对于每个字符,如果等于target,那就改成1,如果不等于target,那就改成0,这样不就可以用这种解法来搞定了吗?我笑了......
所以,这道笔试题的最终解法如下:
n = int(input())
array = list(map(int, input().split()))
target = int(input())
# 魔改array
for i in range(len(array)):
if array[i] == target:
array[i] = 1
else:
array[i] = 0
# 用滑动窗口来解决
def minSwaps(data):
n = len(data)
window_len = sum(data) #所有1的个数
cnt1 = sum(data[ :window_len]) #第一个窗口内1的个数
res = window_len - cnt1 #第一个窗口内0的个数
for R in range(window_len, n): #滑动窗口
cnt1 += data[R] #进R
cnt1 -= data[R - window_len]#弹L
cnt0 = window_len - cnt1 #窗口内0的个数==如果以此窗口的位置为最后结果,需要交换的次数
res = min(res, cnt0) #更新res
return res
print(minSwaps(array))