Here is the contest link.
A. Free Ice Cream for the Kids
To solve this problem efficiently, we simulate the sequence of operations while maintaining two values: the current number of ice cream packs and the count of distressed children. We start with $$$x$$$ packs and iterate through $$$n$$$ operations. If a carrier ($$$+d$$$) arrives, we increase our stock by $$$d$$$. If a child ($$$-d$$$) requests d packs, we check if we have enough ice cream. If yes, we serve the request and decrease our stock; otherwise, we increment the count of distressed kids. By processing all operations sequentially, we determine the final ice cream count and the number of distressed kids. This approach runs in $$$O(n)$$$ time, making it efficient given the constraints.
def solve():
n, cur = map(int, input().split())
distress = 0
for _ in range(n):
op, d = input().split()
d = int(d)
if op == '+':
cur += d
elif cur >= d:
cur -= d
else:
distress += 1
print(cur, distress)
solve()
B. Increasing Sequence
To solve this problem, we need to construct a sequence $$$b$$$ that satisfies the given conditions. The key idea is to ensure that $$$b$$$ is strictly increasing and does not contain any elements from $$$a$$$.
Steps to Solve:
- Initialize a variable
Last_elem
to $$$0$$$, which will keep track of the last value added to the sequence $$$b$$$. - Iterate through each element in the sequence $$$a$$$.
- For each element $$$a_i$$$:
- Increment
Last_elem
by $$$1$$$. - If
Last_elem
is equal to $$$a_i$$$, incrementLast_elem
by $$$1$$$ again to ensure $$$b_i \neq a_i$$$.
- Increment
- After processing all elements,
Last_elem
will hold the minimum value of $$$b_n$$$.
t = int(input())
for _ in range(t):
n = int(input())
a = list(map(int, input().split()))
last_elem = 0
for num in a:
last_elem += 1
if last_elem == num:
last_elem += 1
print(last_elem)
C. Permutation Minimization
We need the deque data structure to solve this problem. We iterate the given permutation starting from the first element and we append to the left of our deque either if it is empty or the element at the 0-th index of our deque is larger than the current element we are about to add to our deque. Otherwise we append to the end of our deque because appending a larger number than what we currently have at the 0-th index of our deque will definitly make our final answer lexicographically larger.
- Time Complexity: O(n)
- Space Complexity: O(n)
from collections import deque
t = int(input())
for _ in range(t):
n = int(input())
arr = list(map(int, input().split()))
ans = deque()
for num in arr:
if ans and num < ans[0]:
ans.appendleft(num)
else:
ans.append(num)
print(*ans)
D. Creating an a-Good String
Try breaking the problem into smaller subproblems by dividing the string into two halves and ensuring one half follows the rule while the other is recursively solved.
We define a recursive function calc(l, r, c)
, where:
-
l
andr
represent the start and end indices of the current segment. -
c
represents the character we are trying to make this segment follow. - The function returns the minimum number of moves needed to convert the substring
s[l:r]
into a c-good string.
Base Case: If the length of the segment is 1 (r = l
):
- If
s[l] = c
,return 0
(no changes needed). - Otherwise,
return 1
(we must changes[l]
toc
).
Recursive Case: We split the segment into two halves and consider two possible transformations:
- case 1: The first half should be fully filled with c, and the second half should be (c+1)-good.
- case 2: The second half should be fully filled with c, and the first half should be (c+1)-good.
For each case, we compute the number of changes needed:
-
left_result
: Changes required to make the first half fully filled with c+
recursive call on the second half to make it (c+1)-good. -
right_result
: Changes required to make the second half fully filled with c+
recursive call on the first half to make it (c+1)-good.
The final result for this segment is: min(left_result,right_result)
Time Complexity The function processes each segment by counting characters (in $$$O(n)$$$ time) and then recursively splitting it into two halves. Since each level of recursion does a total of $$$O(n)$$$ work and there are $$$\log n$$$ levels (because the segment size halves each time), the total complexity is $$$O(n \log n)$$$.
Space complexity The only space used in this solution is the call stack due to recursion. Since the recursion divides the problem into two halves at each step, the maximum depth of the recursion tree is $$$\log n$$$. Therefore, the space complexity is determined by the depth of the recursion, which is $$$O(\log n)$$$.
def calc(l, r, c):
if l == r:
if s[l] == c:
return 0
return 1
mid = (l + r) // 2
left_changes = (mid - l + 1) - s[l:mid + 1].count(c)
left_result = left_changes + calc(mid + 1, r, chr(ord(c) + 1))
right_changes = (r - mid) - s[mid + 1:r + 1].count(c)
right_result = right_changes + calc(l, mid, chr(ord(c) + 1))
return min(left_result, right_result)
for _ in range(int(input())):
n = int(input())
s = input()
print(calc(0, len(s) - 1, 'a'))
E. Strange Mirroring
Instead of building the whole transformed string, find a way to track where each character comes from in the original string. Notice that each step doubles the string and follows a clear pattern.
The problem asks us to find the $$$k^{th}$$$ character after repeatedly transforming a string. Since the string grows extremely fast, it is impossible to generate the final version directly. Instead, we need a smarter way to determine the answer.
Each transformation doubles the string: the first half stays the same, and the second half is the same as the first but with uppercase and lowercase letters swapped. This means that any character in the final string can be traced back to its original position in the first version.
To find the character at position $$$k$$$, we work backwards to see where it came from. If $$$k$$$ is in the first half of a transformation step, it remains unchanged. If it is in the second half, we can shift it back to the first half of the previous step and apply a case swap. We repeat this process until we reach the original string.
For the time complexity, each query runs in $$$O(\log k)$$$, which is at most $$$60$$$ for $$$k \leq 10^{18}$$$. Since $$$q \leq 2 \times 10 ^ 5$$$, this is fast enough.
For the space complexity, the solution uses $$$O(\log k)$$$ extra space since we do not store the full transformed string. Our only space consumption is the call stack from the recursive calls.
import sys
def input():
return sys.stdin.readline().strip()
def read_list():
return list(map(int, input().split()))
# Recursive function to find the character at position k in the transformed sequence
def find_char(depth, k, m, original):
if depth == 1: # Base case: when depth is 1, return the original character
return original[m]
# Calculate the halfway point in the current sequence
half_length = 2 ** (depth - 2)
if k > half_length:
# If k is in the second half, move to the previous depth and swap case
return find_char(depth - 1, k - half_length, m, original).swapcase()
# If k is in the first half, continue in the same depth
return find_char(depth - 1, k, m, original)
def solve():
original_str = input() # Read the original string
_ = int(input())
queries = read_list() # Read the list of queries
n = len(original_str) # Get the length of the original string
results = []
for k in queries:
k -= 1 # Convert to zero-based index
depth = k // n + 1 # Determine the depth level
index = k % n # Find the corresponding index in the original string
results.append(find_char(61, depth, index, original_str)) # Compute the result
print(*results) # Print all results space-separated
t = int(input())
for _ in range(t):
solve()