快速排序(模板)
import java.util.*;
public class Main{
static int[] q = new int[100010];
static int n;
static void quick_sort(int l, int r){
if(l >= r){
return;
}
int i = l - 1, j = r + 1, x = q[i + j >> 1];
while(i < j){
do i ++;while(q[i] < x);
do j --;while(q[j] > x);
if(i < j) {
int t = q[i];
q[i] = q[j];
q[j] = t;
}
}
quick_sort(l, j);
quick_sort(j + 1, r);
}
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
for(int i = 0; i < n; i ++){
q[i] = sc.nextInt();
}
quick_sort(0, n - 1);
for(int i = 0; i < n; i ++){
System.out.print(q[i] + " ");
}
}
}
算法时间复杂度O(nlogn)
例题(求第k个数)
题目
给定一个长度为 n 的整数数列,以及一个整数 k,请用快速选择算法求出数列从小到大排序后的第 k 个数。
输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数(所有整数均在 1∼109 范围内),表示整数数列。
输出格式
输出一个整数,表示数列的第 k 小数。
import java.util.*;
public class Main{
static int[] q = new int[100010];
static int n, k;
static int quick_sort(int l, int r, int k){
if(l == r) return q[l];
int i = l - 1, j = r + 1, x = q[l + r >> 1];
while(i < j){
while(q[++ i] < x);
while(q[-- j] > x);
if(i < j){
int t = q[i];
q[i] = q[j];
q[j] = t;
}
}
int sl = j - l + 1;
if(k <= sl){
return quick_sort(l, j, k);
}
return quick_sort(j + 1, r, k - sl);
}
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
k = sc.nextInt();
for(int i = 0; i < n; i ++) q[i] = sc.nextInt();
int res = quick_sort(0, n - 1, k);
System.out.println(res);
}
}
算法时间复杂度O(n)
归并排序(模板)
import java.util.*;
public class Main{
static int n;
static int[] q = new int[100010];
static int[] temp = new int[100010];
static void merge(int l,int mid, int r){
if(l >= r){
return;
}
int k = 0, i = l, j = mid + 1;
while(i <= mid && j <= r){
if(q[i] < q[j]) temp[k ++] = q[i ++];
else temp[k ++] = q[j ++];
}
while(i <= mid){
temp[k ++] = q[i ++];
}
while(j <= r){
temp[k ++] = q[j ++];
}
for(i = 0; i < k; i ++){
q[l + i] = temp[i];
}
}
static void merge_sort(int l, int r){
if(l < r){
int mid = l + r >> 1;
merge_sort(l, mid);
merge_sort(mid + 1, r);
merge(l, mid, r);
}
}
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
for(int i = 0; i < n; i ++){
q[i] = sc.nextInt();
}
merge_sort(0, n - 1);
for(int i = 0; i < n; i ++){
System.out.print(q[i] + " ");
}
}
}
算法时间复杂度O(nlogn)
例题(逆序对的数量)
题目
给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
import java.util.*;
public class Main{
static int n;
static long ans;
static long[] q = new long[100010];
static long[] temp = new long[100010];
static void merge(int l, int mid, int r){
int i = l, j = mid + 1, k = 0;
while(i <= mid && j <= r){
if(q[i] > q[j]){
temp[k ++] = q[j ++];
ans += mid - i + 1;
}
else temp[k ++] = q[i ++];
}
while(i <= mid){
temp[k ++] = q[i ++];
}
while(j <= r){
temp[k ++] = q[j ++];
}
for(i = 0; i < k; i ++){
q[l + i] = temp[i];
}
}
static void merge_sort(int l, int r){
if(l < r){
int mid = l + r >> 1;
merge_sort(l, mid);
merge_sort(mid + 1, r);
merge(l, mid, r);
}
}
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
for(int i = 0; i < n; i ++){
q[i] = sc.nextInt();
}
merge_sort(0, n - 1);
System.out.println(ans);
}
}