-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path4.py
More file actions
40 lines (36 loc) · 1.14 KB
/
4.py
File metadata and controls
40 lines (36 loc) · 1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
def main():
t = int(input())
for _ in range(t):
n, k = map(int, input().split())
a = list(map(int, input().split()))
a.sort()
m = n - k
intervals = []
for i in range(n - m + 1):
if m % 2 == 1:
median_pos = i + (m - 1) // 2
start = a[median_pos]
end = start
else:
left_pos = i + (m // 2 - 1)
right_pos = i + m // 2
start = a[left_pos]
end = a[right_pos]
intervals.append((start, end))
intervals.sort()
merged = []
for interval in intervals:
if not merged:
merged.append(interval)
else:
last_start, last_end = merged[-1]
current_start, current_end = interval
if current_start <= last_end + 1:
merged[-1] = (last_start, max(last_end, current_end))
else:
merged.append(interval)
total = 0
for start, end in merged:
total += end - start + 1
print(total)
main()