Python Program to Find k maximum elements of array in original order
Last Updated :
28 May, 2022
Given an array arr[] and an integer k, we need to print k maximum elements of given array. The elements should printed in the order of the input.
Note : k is always less than or equal to n.
Examples:
Input : arr[] = {10 50 30 60 15}
k = 2
Output : 50 60
The top 2 elements are printed
as per their appearance in original
array.
Input : arr[] = {50 8 45 12 25 40 84}
k = 3
Output : 50 45 84
Method 1: We search for the maximum element k times in the given array. Each time we find one maximum element, we print it and replace it with minus infinite (-1*sys.maxsize for largest minimum value in Python) in the array. Also, the position of all k maximum elements is marked using an array so that with the help of that array we can print the elements in the order given in the original array. The time complexity of this method is O(n*k).
Python3
# Python program to find k maximum elements
# of array in original order
# import the module
import sys
# Function to print k Maximum elements
def printMax(arr, k, n):
brr=[0]*n
crr=[0]*n
# Coping the array arr
# into crr so that it
# can be used later
for i in range(n):
crr[i]=arr[i]
# Iterating for K-times
for i in range(k):
# Finding the maximum element
# along with its index
maxi=-1*sys.maxsize
for j in range(n):
if(maxi<arr[j]):
maxi=arr[j]
index=j
# Assigning 1 in order
# to mark the position
# of all k maximum numbers
brr[index]=1
arr[index]=-1*sys.maxsize
for i in range(n):
# Printing the k maximum
# elements array
if(brr[i]==1):
print(crr[i],end=" ")
# Driver code
arr = [ 50, 8, 45, 12, 25, 40, 84 ]
n = len(arr)
k = 3;
printMax(arr, k, n)
# This code is contributed by Pushpesh Raj.
Time Complexity: O(n*k)
Auxiliary Space: O(n)
Method 2: In this method, we store the original array in a new array and will sort the new array in descending order. After sorting, we iterate the original array from 0 to n and print all those elements that appear in first k elements of new array. For searching, we can do Binary Search.
Python3
# Python3 program to find k maximum elements
# of array in original order
# Function to print Maximum elements
def printMax(arr, k, n):
# vector to store the copy of the
# original array
brr = arr.copy()
# Sorting the vector in descending
# order. Please refer below link for
# details
brr.sort(reverse = True)
# Traversing through original array and
# print all those elements that are
# in first k of sorted vector.
for i in range(n):
if (arr[i] in brr[0:k]):
print(arr[i], end = " ")
# Driver code
arr = [ 50, 8, 45, 12, 25, 40, 84 ]
n = len(arr)
k = 3
printMax(arr, k, n)
# This code is contributed by SHUBHAMSINGH10
Time Complexity: O(n Log n) for sorting.
Auxiliary Space: O(n)
Please refer complete article on Find k maximum elements of array in original order for more details!
Similar Reads
Javascript Program to Find k maximum elements of array in original order Given an array arr[] and an integer k, we need to print k maximum elements of given array. The elements should printed in the order of the input.Note: k is always less than or equal to n.Examples: Input : arr[] = {10 50 30 60 15} k = 2 Output : 50 60 The top 2 elements are printed as per their appea
3 min read
Find k maximum elements of array in original order Given an array arr[] and an integer k, we need to print k maximum elements of given array. The elements should printed in the order of the input.Note : k is always less than or equal to n. Examples: Input : arr[] = {10 50 30 60 15} k = 2 Output : 50 60 The top 2 elements are printed as per their app
11 min read
Python Program to Find Largest Element in an Array To find the largest element in an array, iterate over each element and compare it with the current largest element. If an element is greater, update the largest element. At the end of the iteration, the largest element will be found. Given an array, find the largest element in it. Input : arr[] = {1
4 min read
Find the maximum and minimum element in a NumPy array An array can be considered as a container with the same types of elements. Python has its array module named array. We can simply import the module and create our array. But this module has some of its drawbacks. The main disadvantage is we can't create a multidimensional array. And the data type mu
4 min read
Maximize the minimum value of Array by performing given operations at most K times Given array A[] of size N and integer K, the task for this problem is to maximize the minimum value of the array by performing given operations at most K times. In one operation choose any index and increase that array element by 1. Examples: Input: A[] = {3, 1, 2, 4, 6, 2, 5}, K = 8Output: 4Explana
10 min read