题目
题目链接:
https://www.nowcoder.com/practice/41796daa4c7e4e5ab984b2c16c24a1de
思路
双指针
参考答案C++
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param arr1 int整型vector
* @param arr2 int整型vector
* @param target int整型
* @return int整型
*/
int findKthNum(vector<int>& arr1, vector<int>& arr2, int target) {
//双指针
int n = arr1.size();
int m = arr2.size();
int p1 = 0, p2 = 0, cnt = 0, ans = 0;
while (p1 < n && p2 < m) {
if (arr1[p1] >= arr2[p2]) {
ans = arr2[p2++];
} else {
ans = arr1[p1++];
}
cnt++;
if (cnt == target) {
return ans;
}
}
while (p1 < n) {
ans = arr1[p1++];
cnt++;
if (cnt == target) {
return ans;
}
}
while (p2 <m) {
ans = arr2[p2++];
cnt++;
if (cnt == target) {
return ans;
}
}
return ans;
}
};
参考答案Java
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param arr1 int整型一维数组
* @param arr2 int整型一维数组
* @param target int整型
* @return int整型
*/
public int findKthNum (int[] arr1, int[] arr2, int target) {
//双指针
int n = arr1.length, m = arr2.length;
int p1 = 0, p2 = 0, cnt = 0;
int ans = 0;
while (p1 < n && p2 < m) {
if (arr1[p1] >= arr2[p2]) {
ans = arr2[p2++];
} else {
ans = arr1[p1++];
}
cnt++;
if (cnt == target) {
return ans;
}
}
while (p1 < n) {
ans = arr1[p1++];
cnt++;
if (cnt == target) {
return ans;
}
}
while (p2 < m) {
ans = arr2[p2++];
cnt++;
if (cnt == target) {
return ans;
}
}
return ans;
}
}
参考答案Go
package main
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param arr1 int整型一维数组
* @param arr2 int整型一维数组
* @param target int整型
* @return int整型
*/
func findKthNum(arr1 []int, arr2 []int, target int) int {
// 双指针
n := len(arr1)
m := len(arr2)
p1 := 0
p2 := 0
cnt := 0
ans := 0
for p1 < n && p2 < m {
if arr1[p1] >= arr2[p2] {
ans = arr2[p2]
p2++
} else {
ans = arr1[p1]
p1++
}
cnt++
if cnt == target {
return ans
}
}
for p1 < n {
ans = arr1[p1]
p1++
cnt++
if cnt == target {
return ans
}
}
for p2 < m {
ans = arr2[p2]
p2++
cnt++
if cnt == target {
return ans
}
}
return ans
}