chore(Java): add k-th element of 2 sorted arrays (#366)

pull/367/head
Aayush 2021-06-22 21:47:02 +05:30 committed by GitHub
parent 3e2c18ed46
commit ca33830b3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 77 additions and 0 deletions

View File

@ -8,6 +8,7 @@
4. [Unique Digits of Large Number](arrays/unique-digits-of-large-number.java) 4. [Unique Digits of Large Number](arrays/unique-digits-of-large-number.java)
5. [Majority Element](arrays/majority-element.java) 5. [Majority Element](arrays/majority-element.java)
6. [Longest Consecutive Subsequence](arrays/longest-consecutive-subsequence.java) 6. [Longest Consecutive Subsequence](arrays/longest-consecutive-subsequence.java)
7. [K-th Element of Two Sorted Arrays](arrays/kth-element-2sorted-array.java)
## Graphs ## Graphs

View File

@ -0,0 +1,76 @@
//Given two sorted arrays arr1 and arr2 of size M and N respectively and an element K.
//The task is to find the element that would be at the kth position of the final sorted array.
// Time Complexity: O(k)
// Auxiliary Space: O(1)
import java.util.*;
import java.lang.*;
import java.io.*;
class Solution {
public long kthElement( int arr1[], int arr2[], int n, int m, int k) {
int i=0, j=0, curr_k=0;
while(i<n && j<m){
if(arr1[i]<arr2[j]){
curr_k++;
if(k==curr_k) return arr1[i];
i++;
} else {
curr_k++;
if(k==curr_k) return arr2[j];
j++;
}
}
while (i < n) {
curr_k++;
if (k == curr_k)
return arr1[i];
i++;
}
while (j < m) {
curr_k++;
if (k == curr_k)
return arr2[j];
j++;
}
return -1;
}
}
class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int t = Integer.parseInt(br.readLine().trim());
while(t-->0) {
StringTokenizer stt = new StringTokenizer(br.readLine());
int n = Integer.parseInt(stt.nextToken());
int m = Integer.parseInt(stt.nextToken());
int k = Integer.parseInt(stt.nextToken());
int a[] = new int[(int)(n)];
int b[] = new int[(int)(m)];
String inputLine[] = br.readLine().trim().split(" ");
for (int i = 0; i < n; i++) {
a[i] = Integer.parseInt(inputLine[i]);
}
String inputLine1[] = br.readLine().trim().split(" ");
for (int i = 0; i < m; i++) {
b[i] = Integer.parseInt(inputLine1[i]);
}
Solution obj = new Solution();
System.out.println(obj.kthElement( a, b, n, m, k));
}
}
}
/* Test Case:
Input:
arr1[] = {2, 3, 6, 7, 9}
arr2[] = {1, 4, 8, 10}
k = 5
Output:
6
Explanation:
The final sorted array would be -
1, 2, 3, 4, 6, 7, 8, 9, 10
The 5th element of this array is 6. */