求第k小的数(洛谷P1923题题解,Java/C++语言描述)
生活随笔
收集整理的這篇文章主要介紹了
求第k小的数(洛谷P1923题题解,Java/C++语言描述)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目要求
題目鏈接
分析
《查找第K大/小元素算法》
這題不使用線性復雜度算法是過不去的(除非你用C++的[內置函數]or[快讀+sort()])
Java應該是不可能過去的,你用C++可能卡在T上,Java還沒到T呢,M直接卡爆了,頂多60分水平吧。
下面是Java版本,線性復雜度,盡可能優化,60分:
import java.io.BufferedReader; import java.io.IOException; import java.io.InputStreamReader;/*** 線性復雜度算法* 該算法的C++版本可AC,但Java版本就會MLE/TLE* 該算法C++版本跑的比nth_element()跑得快*/ public class Main {private static int[] nums;private static int k, num;private static void swap(int left, int right) {int temp = nums[left];nums[left] = nums[right];nums[right] = temp;}private static void sort(int left, int right) {int left_mid=left, right_mid=right, mid= nums[(left+right)/2];while (left_mid <= right_mid) {while (nums[right_mid] > mid) {right_mid--;}while (nums[left_mid] < mid) {left_mid++;}if (left_mid <= right_mid && right_mid < num) {swap(left_mid, right_mid);left_mid++;right_mid--;}}if (k <= right_mid) {sort(left, right_mid);} else if (left_mid <= k) {sort(left_mid, right);} else {System.out.println(nums[right_mid+1]);}}public static void main(String[] args) throws IOException {BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));String[] line1 = reader.readLine().split("\\s+");num = Integer.parseInt(line1[0]);k = Integer.parseInt(line1[1]);nums = new int[num];String[] array = reader.readLine().split("\\s+");for (int i = 0; i < num; i++) {nums[i] = Integer.parseInt(array[i]);}reader.close();sort(0, num-1);}}AC代碼(C++)
#include<bits/stdc++.h>using namespace std;long long num, k, nums[5000010];int main() {scanf("%d%d", &num, &k);for(int i=0; i < num; i++) {scanf("%d", &nums[i]);}nth_element(nums, nums + k, nums + num);printf("%d", nums[k]); }總結
以上是生活随笔為你收集整理的求第k小的数(洛谷P1923题题解,Java/C++语言描述)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【离散数学】代数系统分类
- 下一篇: 【计算机科学基础】二进制加减法不用原码的