Find all indices of a given element in sorted form of given Array
Last Updated :
19 Jul, 2022
Given an array arr[] of integers of size N and a target value val. Your task is to find the indices of val in the array after sorting the array in increasing order.
Note: The indices must be in increasing order.
Examples:
Input: arr = [1, 2, 5, 2, 3], val = 2
Output: 1 2
Explanation: After sorting, arr[] becomes [1, 2, 2, 3, 5]. The indices where arr[i] = 2 are 1 and 2. As the indices should be in increasing order, that's why they are (1, 2) and not (2, 1)
Input: arr = [1, 2, 5, 2, 3], val = 6
Output: []
Explanation: After sorting, nums is [1, 2, 2, 3, 5]. The value 6 is not present in the array.
Naive Approach: The concept of this approach is based on sorting. The array is sorted in increasing order. Then the sorted array is traversed. While traversing the array if any value matches the target, that index is added to the answer list. After the iteration is complete the answer list is returned.
Time Complexity: O(N*logN)
Auxiliary Space: O(1)
Efficient Approach: This approach is based on the observation of a sorted array. In an array which is sorted in increasing order all the elements before val are less than val. So, to get the indices of val in a sorted array, instead of performing sort operation, we can simply count the elements less than val. If the count is x then val will start from x-th index (because of 0-based indexing). Follow the steps mentioned below:
- Traverse the array. Use variables to store the count of elements less than val (smallCount) and elements having a value same as val (sameCount).
- If the value of an element is less than val increment smallCount by one.
- If the value is same as val increment sameCount by one.
- After traversal is complete, the indices will start from smallCount and end at (smallCount+sameCount-1)
Below is the implementation of the above approach:
C++
// C++ implementation of the above approach
#include <bits/stdc++.h>
using namespace std;
// Function to find indices
// of val in array after sorting
vector<int> targetIndices(vector<int>& nums,
int val)
{
int count_less = 0;
int count_target = 0;
// Loop to count smaller elements and val
for (auto& it : nums) {
if (it == val)
count_target++;
if (it < val)
count_less++;
}
// List to store indices
vector<int> ans;
while (count_target--) {
ans.push_back(count_less++);
}
return ans;
}
// Driver code
int main()
{
vector<int> nums{ 1, 2, 5, 2, 3 };
int val = 2;
vector<int> ans = targetIndices(nums, val);
// Loop to print indices
for (int i = 0; i < ans.size(); i++) {
cout << ans[i] << " ";
}
return 0;
}
Java
// Java implementation of the above approach
import java.util.*;
public class GFG {
// Function to find indices
// of val in array after sorting
static List<Integer> targetIndices(int[] nums, int val)
{
int count_less = 0;
int count_target = 0;
// Loop to count smaller elements and val
for (int i = 0; i < nums.length; i++) {
if (nums[i] == val)
count_target++;
if (nums[i] < val)
count_less++;
}
// List to store indices
List<Integer> ans = new ArrayList<Integer>();
while (count_target > 0) {
ans.add(count_less++);
count_target--;
}
return ans;
}
// Driver code
public static void main(String args[])
{
int[] nums = { 1, 2, 5, 2, 3 };
int val = 2;
List<Integer> ans = targetIndices(nums, val);
// Loop to print indices
for (int i = 0; i < ans.size(); i++) {
System.out.print(ans.get(i) + " ");
}
}
}
// This code is contributed by Samim Hossain Mondal.
Python3
# Python implementation of the above approach
# Function to find indices
# of val in array after sorting
def targetIndices(nums, val):
count_less = 0
count_target = 0
# Loop to count smaller elements and val
for i in range(len(nums)):
if (nums[i] == val):
count_target += 1
if (nums[i] < val):
count_less += 1
# List to store indices
ans = []
while (count_target):
ans.append(count_less)
count_less += 1
count_target -= 1
return ans
# Driver code
nums = [1, 2, 5, 2, 3]
val = 2
ans = targetIndices(nums, val)
# Loop to print indices
for i in range(len(ans)):
print(ans[i], end=" ")
# This code is contributed by Saurabh Jaiswal
C#
// C# implementation of the above approach
using System;
using System.Collections;
using System.Collections.Generic;
class GFG {
// Function to find indices
// of val in array after sorting
static ArrayList targetIndices(int[] nums, int val)
{
int count_less = 0;
int count_target = 0;
// Loop to count smaller elements and val
for (int i = 0; i < nums.Length; i++) {
if (nums[i] == val)
count_target++;
if (nums[i] < val)
count_less++;
}
// List to store indices
ArrayList ans = new ArrayList();
while (count_target > 0) {
ans.Add(count_less++);
count_target--;
}
return ans;
}
// Driver code
public static void Main()
{
int[] nums = { 1, 2, 5, 2, 3 };
int val = 2;
ArrayList ans = targetIndices(nums, val);
// Loop to print indices
for (int i = 0; i < ans.Count; i++) {
Console.Write(ans[i] + " ");
}
}
}
// This code is contributed by Samim Hossain Mondal.
JavaScript
<script>
// Javascript implementation of the above approach
// Function to find indices
// of val in array after sorting
function targetIndices(nums, val)
{
let count_less = 0;
let count_target = 0;
// Loop to count smaller elements and val
for (let i = 0; i < nums.length; i++) {
if (nums[i] == val)
count_target++;
if (nums[i] < val)
count_less++;
}
// List to store indices
let ans = [];
while (count_target--) {
ans.push(count_less++);
}
return ans;
}
// Driver code
let nums = [ 1, 2, 5, 2, 3 ];
let val = 2;
let ans = targetIndices(nums, val);
// Loop to print indices
for (let i = 0; i < ans.length; i++) {
document.write(ans[i] + " ");
}
// This code is contributed by Samim Hossain Mondal.
</script>
Time Complexity: O(N)
Auxiliary Space: O(1)
Similar Reads
DSA Tutorial - Learn Data Structures and Algorithms DSA (Data Structures and Algorithms) is the study of organizing data efficiently using data structures like arrays, stacks, and trees, paired with step-by-step procedures (or algorithms) to solve problems effectively. Data structures manage how data is stored and accessed, while algorithms focus on
7 min read
Quick Sort QuickSort is a sorting algorithm based on the Divide and Conquer that picks an element as a pivot and partitions the given array around the picked pivot by placing the pivot in its correct position in the sorted array. It works on the principle of divide and conquer, breaking down the problem into s
12 min read
Merge Sort - Data Structure and Algorithms Tutorials Merge sort is a popular sorting algorithm known for its efficiency and stability. It follows the divide-and-conquer approach. It works by recursively dividing the input array into two halves, recursively sorting the two halves and finally merging them back together to obtain the sorted array. Merge
14 min read
Bubble Sort Algorithm Bubble Sort is the simplest sorting algorithm that works by repeatedly swapping the adjacent elements if they are in the wrong order. This algorithm is not suitable for large data sets as its average and worst-case time complexity are quite high.We sort the array using multiple passes. After the fir
8 min read
Data Structures Tutorial Data structures are the fundamental building blocks of computer programming. They define how data is organized, stored, and manipulated within a program. Understanding data structures is very important for developing efficient and effective algorithms. What is Data Structure?A data structure is a st
2 min read
Breadth First Search or BFS for a Graph Given a undirected graph represented by an adjacency list adj, where each adj[i] represents the list of vertices connected to vertex i. Perform a Breadth First Search (BFS) traversal starting from vertex 0, visiting vertices from left to right according to the adjacency list, and return a list conta
15+ min read
Binary Search Algorithm - Iterative and Recursive Implementation Binary Search Algorithm is a searching algorithm used in a sorted array by repeatedly dividing the search interval in half. The idea of binary search is to use the information that the array is sorted and reduce the time complexity to O(log N). Binary Search AlgorithmConditions to apply Binary Searc
15 min read
Insertion Sort Algorithm Insertion sort is a simple sorting algorithm that works by iteratively inserting each element of an unsorted list into its correct position in a sorted portion of the list. It is like sorting playing cards in your hands. You split the cards into two groups: the sorted cards and the unsorted cards. T
9 min read
Dijkstra's Algorithm to find Shortest Paths from a Source to all Given a weighted undirected graph represented as an edge list and a source vertex src, find the shortest path distances from the source vertex to all other vertices in the graph. The graph contains V vertices, numbered from 0 to V - 1.Note: The given graph does not contain any negative edge. Example
12 min read
Selection Sort Selection Sort is a comparison-based sorting algorithm. It sorts an array by repeatedly selecting the smallest (or largest) element from the unsorted portion and swapping it with the first unsorted element. This process continues until the entire array is sorted.First we find the smallest element an
8 min read