牛客NC251 多数组第 K 小数【 中等 双指针 C++、Java、Go】

题目

题目链接：
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
}