2113. Elements in Array After Removing and Replacing Elements
Problem Description
The problem presents an array nums
representing a sequence of integers and a dynamic process that alternately removes elements from the beginning of the array and restores them after the array has been emptied. This process repeats indefinitely. Explicitly, each minute, one element from the beginning of the array is removed until the array is empty. Subsequently, each minute, an element is appended to the end of the array in the same order they were removed until nums
is restored to its original state.
The problem also provides a set of queries where each query is a pair consisting of a minute time_j
and an index index_j
. The task is to find out what element, if any, exists at the given index at the specific time. If the index at a given time exceeds the current length of the array (since elements may have been removed or not yet restored), the answer to that query is -1
.
Intuition
The key to solving this problem lies in recognizing the cyclic nature of changes to the array. The array completes one cycle of changes every 2 * length(nums)
minutes: half of the cycle for removing elements and the other half for restoring them. Thus, we need to determine the state of the array at a given point in this cycle to answer the queries.
Firstly, since the array repeats the same pattern every 2 * n
minutes (n
being the initial length of nums
), we can simplify each query time t
by using the modulus operator t % (2 * n)
to find the equivalent time within the first cycle.
Next, we identify two distinct phases within a cycle:
-
The removal phase: this lasts for the first
n
minutes of the cycle. In this phase, we can simply check whether the indexi
specified in the query is still within the bounds of the array aftert
elements are removed from the start. The current element can be directly accessed asnums[i + t]
. -
The restoration phase: this starts after the removal phase and lasts until the end of the cycle (from minute
n
to minute2 * n
). During this phase, we can find out if the element ati
has been restored by checking ifi < t - n
(which means that the element at indexi
has already been appended back). In this case, the element at indexi
is simplynums[i]
.
Combining these phases, we iterate over each query, compute the effective time and determine the answer accordingly. The insight to solve this problem efficiently is to understand the cyclical process and calculate the element position relative to the current phase of the cycle.
Solution Approach
To implement the solution, we must navigate through the queries to determine the values at specific times. Here’s a step-by-step approach that is followed by the given solution:
-
Initialization: We initialize an array
ans
with the same number of elements as queries filled with-1
. This array will store the answers to each query. The variablen
holds the length ofnums
, andm
holds the number of queries. -
Query Simplification: For each query consisting of time
t
and indexi
, we first reducet
to its equivalent time within the first cycle by calculatingt %= 2 * n
. This is crucial since the array's behavior repeats every2 * n
minutes. It avoids unnecessary repetition by focusing on a single cycle. -
Handling Queries:
- We loop through each query using
j
as the index and unpack each query intot
andi
. - We check if the current minute
t
falls within the removal phase (t < n
). If true, we see if the queried indexi
is valid after removingt
elements from the beginning. The checki < n - t
assures there is still an element at the index after the removal. If this is the case, we retrieve the valuenums[i + t]
. - If the current minute
t
falls within the restoration phase (t > n
), we perform a different check:i < t - n
. This checks if the element at indexi
has been restored by this minute. If it has, we know that the element at indexi
is the same asnums[i]
since we restore the elements in the same order they were removed.
- We loop through each query using
These steps use simple iterations and conditions to determine each query's answer effectively. The algorithm's time complexity is O(m), as each query is handled in constant time, given that we have precomputed n
(the length of nums
). There are no complex data structures used apart from an array to store the results.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Let's walk through an example to illustrate the solution approach. Suppose we have an array nums = [1,2,3,4]
and the queries [(3, 1), (0, 2), (5, 0), (10, 1)]
.
-
Initialization: Initialize an array to store answers
ans = [-1, -1, -1, -1]
(since we have four queries). Also, we haven = 4
(the length ofnums
) andm = 4
(the number of queries). -
Query Simplification:
- For the first query
(3, 1)
:- We simplify
t = 3 % (2 * 4) = 3
. The new query is(3, 1)
.
- We simplify
- For the second query
(0, 2)
:- Simplification is not needed since
t = 0
. The query remains(0, 2)
.
- Simplification is not needed since
- For the third query
(5, 0)
:t = 5 % (2 * 4) = 5
. The new query is(5, 0)
.
- For the fourth query
(10, 1)
:t = 10 % (2 * 4) = 2
. The new query is(2, 1)
.
- For the first query
-
Handling Queries:
- For
(3, 1)
: We are in the removal phase because3 < 4
. Is index1
valid after removing 3 elements?1 < 4 - 3
, yes. Soans[0] = nums[1 + 3] = nums[4]
, but4
is outside the index range fornums
, henceans[0]
remains-1
. - For
(0, 2)
: It's the beginning of the cycle. No elements have been removed yet, soans[1] = nums[2] = 3
. - For
(5, 0)
: We are in the restoration phase because5 > 4
. Is index0
restored after 5 minutes when 1 minute of restoration has passed?0 < 5 - 4
, yes, soans[2] = nums[0] = 1
. - For
(2, 1)
: Still in the removal phase as2 < 4
. Is index1
valid after 2 have been removed?1 < 4 - 2
, yes, soans[3] = nums[1 + 2] = nums[3] = 4
.
- For
Final ans
array after evaluating all queries is [-1, 3, 1, 4]
. Each element of ans
correlates to the results of the respective queries, suggesting that at those specific times and indices, these were the correct values in the array.
Solution Implementation
1class Solution:
2 def elementInNums(self, nums: List[int], queries: List[List[int]]) -> List[int]:
3 # Determine the length of the 'nums' list and the length of 'queries'.
4 num_length, query_count = len(nums), len(queries)
5
6 # Initialize the answer list with -1 for each query.
7 answers = [-1] * query_count
8
9 # Process each query and determine the element at the given index if it exists.
10 for index, (shift, target_index) in enumerate(queries):
11 # The actual shift is the remainder when 'shift' is divided by twice the 'num_length',
12 # since a full rotation would bring the array back to the original state.
13 actual_shift = shift % (2 * num_length)
14
15 # Case 1: If the shift is less than the array length and the target index is within the range
16 # after the shift, update the answer.
17 if actual_shift < num_length and target_index < num_length - actual_shift:
18 answers[index] = nums[target_index + actual_shift]
19
20 # Case 2: If the shift is greater than the array length and the target index is within
21 # the range before the shift, update the answer.
22 elif actual_shift > num_length and target_index < actual_shift - num_length:
23 answers[index] = nums[target_index]
24
25 # After processing all queries, return the answers list.
26 return answers
27
1class Solution {
2
3 /**
4 * Finds elements in the array after performing certain query operations.
5 *
6 * @param nums The input array
7 * @param queries Instructions for each query, containing two values [t, i]
8 * @return Array of elements according to each query
9 */
10 public int[] elementInNums(int[] nums, int[][] queries) {
11 int numLength = nums.length; // the length of the input array
12 int queryCount = queries.length; // the number of queries
13 int[] result = new int[queryCount]; // array to store the result elements
14
15 // Loop over each query
16 for (int j = 0; j < queryCount; ++j) {
17 result[j] = -1; // initialize the result as -1 (not found)
18 int period = queries[j][0], index = queries[j][1]; // extract period (t) and index (i) from the query
19 period %= (2 * numLength); // reduce the period within the range of 0 to 2*n
20
21 // Check if the required element falls within the unshifted part of the array
22 if (period < numLength && index < numLength - period) {
23 result[j] = nums[index + period]; // set the element after 'period' shifts
24 }
25 // Check if the required element falls within the shifted part of the array
26 else if (period > numLength && index < period - numLength) {
27 result[j] = nums[index]; // set the element without shift as it has cycled back
28 }
29 }
30
31 return result; // return the result array with elements according to queries
32 }
33}
34
1#include <vector>
2
3class Solution {
4public:
5 // Function to get the elements in nums after applying rotations from queries
6 std::vector<int> elementInNums(std::vector<int>& nums, std::vector<std::vector<int>>& queries) {
7
8 // n is the size of the nums vector; m is the size of the queries vector
9 int n = nums.size(), m = queries.size();
10
11 // Initialize an answer vector with size m and fill it with -1
12 std::vector<int> ans(m, -1);
13
14 // Iterate over each query
15 for (int j = 0; j < m; ++j) {
16
17 // t is the number of rotations and i is the index for the current query
18 int t = queries[j][0], i = queries[j][1];
19
20 // Since the array rotates back to the original position after n*2 rotations,
21 // we use modulo to simplify the number of rotations
22 t %= (n * 2);
23
24 // If the number of rotations is less than the size of nums
25 // and the index is within the bounds after the rotation, get the element
26 if (t < n && i < n - t) {
27 ans[j] = nums[i + t];
28 }
29 // If the number of rotations is more than size but the index is within bounds,
30 // that means the array has been rotated back to its original position
31 // and possibly further, we get the element from the original position
32 else if (t >= n && i < t - n) {
33 ans[j] = nums[i];
34 }
35 // If none of the above conditions meet, the answer remains -1
36 // which indicates that the element is not accessible after the rotation
37 }
38
39 // Return the answer vector
40 return ans;
41 }
42};
43
1// Importing array type from Typescript for type annotations
2import { Array } from "typescript";
3
4// Function to get the elements in nums after applying rotations from queries
5function elementInNums(nums: Array<number>, queries: Array<Array<number>>): Array<number> {
6
7 // n is the size of the nums array; m is the size of the queries array
8 const n: number = nums.length;
9 const m: number = queries.length;
10
11 // Initialize an answer array with size m and fill it with -1
12 let ans: Array<number> = new Array(m).fill(-1);
13
14 // Iterate over each query
15 for (let j = 0; j < m; ++j) {
16
17 // t is the number of rotations and i is the index for the current query
18 let t: number = queries[j][0];
19 const i: number = queries[j][1];
20
21 // Since the array rotates back to the original position after n * 2 rotations,
22 // we use modulo to simplify the number of rotations
23 t %= (n * 2);
24
25 // If the number of rotations is less than the size of nums
26 // and the index is within the bounds after the rotation, get the element
27 if (t < n && i < n - t) {
28 ans[j] = nums[i + t];
29 }
30 // If the number of rotations is more than the size but the index is within bounds,
31 // that means the array has been rotated back to its original position
32 // and possibly further, we get the element from the original position
33 else if (t >= n && i < t - n) {
34 ans[j] = nums[i];
35 }
36 // If none of the above conditions meet, the answer remains -1
37 // which indicates that the element is not accessible after the rotation
38 }
39
40 // Return the answer array
41 return ans;
42}
43
Time and Space Complexity
The code provided involves iterating over the list of queries and computing the answer for each query based on given conditions. We'll analyze both the time complexity and space complexity.
Time Complexity
The time complexity of the algorithm depends on the number of queries we need to process, as the iterations inside the loop do not depend on the size of the nums
list. Since we are iterating over every single query once, the time complexity is O(m)
, where m
is the number of queries.
We do not have any nested loops that iterate over the length of the nums
list, and the modulo and if-else operations inside the for loop are all constant time operations.
Hence, the overall time complexity is O(m)
.
Space Complexity
The space complexity is determined by the additional space required by the program which is not part of the input. Here, the main additional space used is the ans
list, which contains one element for each query.
Since we create an ans
list of size m
, the space complexity is O(m)
, where m
is the number of queries.
We do not use any additional data structures that grow with the size of the input, so there is no further space complexity to consider.
Therefore, the overall space complexity is also O(m)
.
Learn more about how to find time and space complexity quickly using problem constraints.
Which data structure is used in a depth first search?
Recommended Readings
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Recursion Recursion is one of the most important concepts in computer science Simply speaking recursion is the process of a function calling itself Using a real life analogy imagine a scenario where you invite your friends to lunch https algomonster s3 us east 2 amazonaws com recursion jpg You first
Runtime Overview When learning about algorithms and data structures you'll frequently encounter the term time complexity This concept is fundamental in computer science and offers insights into how long an algorithm takes to complete given a certain input size What is Time Complexity Time complexity represents the amount of time
Want a Structured Path to Master System Design Too? Don’t Miss This!