刷题记录acwing

 第k个数

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式
第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在 1∼1e9 范围内），表示整数数列。

输出格式
输出一个整数，表示数列的第 k 小数。

数据范围
1≤n≤100000,
1≤k≤n
输入样例：
5 3
2 4 1 5 3
输出样例：
3

分析：这题涉及到了快速选择算法，当然我们也可以通过快速排序后输出对应的数组下表，但是快速排序算法的时间复杂度为o（nlogn），而快速选择算法的时间复杂度为o（n），与快速排序相同，我们需要找到一个基准数x，可以是左边也可以是右边,中间。然后使x左边的数小于等于x，x右边的数大于等于x，我们假设x左边的数个数为Sl，x右边的数的个数为Sr，那么如果k小于Sl，则说明第k个数字在做左边，我们对左区间进行递归，如果k大于Sl，说明k在右边，且在右边的第k-Sl

 

#include <iostream>
using namespace std;
const int N=1e5+10;
int q[N];
int quick_find(int l,int r,int k)
{
	if(l==r) return q[l];
	int x=q[l],i=l-1,j=r+1;
	while(i<j)
	{
		while(q[++i]<x);
		while(q[--j]>x);
		if(i<j) swap(q[i],q[j]);
	}
	int sl=j-l+1;
	if(k<=sl) return quick_find(l,j,k); 
	else 
	return quick_find(j+1,r,k-sl);
}
int main()
{
	int n,k;
	cin>>n>>k;
	for(int i=0;i<n;i++) cin>>q[i];
	cout<<quick_find(0,n-1,k);
	return 0;
} 

逆序对的数量

题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。

最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列有，a[i]>a[j],而i<j，则称这两个元素为一个逆序对。。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。

Update:数据已加强。

输入格式

第一行，一个数 n，表示序列中有 n个数。

第二行 n 个数，表示给定的序列。序列中每个数字不超过 10^9

输出格式

输出序列中逆序对的数目。

输入输出样例

输入 #1

6
5 4 2 6 3 1

输出 #1

11

逆序对使用分治的方法解决，相当于是进行一次归并排序。因此时间复杂度为nlogn ，对于这题，我们有可能出现逆序对的地方可能在【l，mid】也可能在【mid+1，r】或者出现在mid两侧的区间中，所以我们只要把这三个区间的逆序对数量分别统计出来，相加就可以得到答案

#include <iostream>
using namespace std;
typedef long long LL;
const int N=1e6+10;
int temp[N],q[N];
LL merge_sort(int l,int r )
{
	if(l>=r) return 0;
	int mid=l+r>>1;
	LL res=merge_sort(l,mid)+merge_sort(mid+1,r);
	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++];
		res+=mid-i+1;
	}
	while(i<=mid) temp[k++]=q[i++];
	while(j<=r) temp[k++]=q[j++];
	//把temp的值放到q里去
	for(int i=l,j=0;i<=r;i++,j++) q[i]=temp[j];
	return res ;
}
int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++) cin>>q[i];
	cout<<merge_sort(0,n-1);
	return 0;
}