洛谷 P4447 [AHOI2018初中组] 分组 python解析

P4447 [AHOI2018初中组] 分组

时间：2023.10.16
题目地址：[AHOI2018初中组] 分组

题目分析

刚开始拿到这个题目，想想简单了，我的想法是先排序，然后一个个进，判断是否连续，不连续了就换个新列表存，记录此时组的长度，更新最小值。很显然没有考虑到如下例子，就是重复的时候。只有60分。（如果有跟我一样的情况，试试下面两个样例，过了的话就能突破到100分了。）


随后就对代码进行了改进那就是要拿个列表将每次的组的长度和最后一个数存起来，每次拿到一个数，对这些组遍历，找到可以插入的组（这里就要考虑贪心了，从后向前搜索，因为要保证人数最少组的人数最小，看不明白的话，就用自己拿上面的样例，试一下，从前搜索和从后的区别。）
100分拿下了，很显然#11超时，是后来加的，那么，题目还要要求你用二分来搜索。

最后就是在搜索的时候用二分找最右边的下标就行了。
二分不熟的可以看看我之前发的就是二分找下标，去练练二分。
洛谷 P2249 【深基13.例1】查找 python解析

代码

1. 60分代码

n = int(input())
nums = [int(i) for i in input().split()]

nums.sort()

minn = float('inf')
state = []
for i in range(n):
    if len(state) == 0:
        state.append(nums[i])
        continue
    if nums[i] == state[-1] + 1:
        state.append(nums[i])
        continue
    # 找不到了，另开一组，并更新最小值（未考虑）
    minn = min(minn, len(state))
    state = [nums[i]]
minn = min(minn, len(state))
    
print(minn)

2. 100分代码

n = int(input())
nums = [int(i) for i in input().split()]

nums.sort()

state = [[1, nums[0]]] # 存每一个组，一个组又包含组的长度和最后一个数
for i in range(1, n):
    for j in range(len(state)-1, -1, -1):  # 贪心体现
        if nums[i] == state[j][1] + 1:
            state[j][0] += 1
            state[j][1] += 1
            break
    else: # 找不到就另开一组
        state.append([1, nums[i]])
# 取最小长度
minn = min([temp[0] for temp in state])
print(minn)

3. AC代码

n = int(input())
nums = [int(i) for i in input().split()]

nums.sort()

state = [[1, nums[0]]]
for i in range(1, n):
    left = 0
    right = len(state) - 1
    last = -1
    while left <= right:
        mid = (left + right) // 2
        if state[mid][1] > nums[i] - 1:
            right = mid - 1
        elif state[mid][1] < nums[i] - 1:
            left = mid + 1
        else:
        	# last记录下标
            last = mid
            # 接着向右找
            left = mid + 1
    # 找到了
    if last != -1:
        state[last][0] += 1
        state[last][1] += 1
    # 找不到
    else:
        state.append([1, nums[i]])
# 取最小长度
minn = min([temp[0] for temp in state])
print(minn)