数的查找

一.第k小的数在数组中的位置

方法：

（1）在数组a五个数为一组，分为[n/5]个组。

（2）对[n/5]个组的数进行组内排序。

（3）选择每组的中位数，将这些中位数全部移动到数组的前面。

（4）对中位数组进行排序，选择中位数组中的中位数x作为划分基准。

（5）以x为基准进行二分，比较，递归寻找即可

时间复杂度最差为O（n）

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int a[100005],b[100005];
int i,j,k,m,l;
int cmp(int a,int b)
{
    return a<b;
}
void Swap()
{
    swap(a[i],a[j]);
    swap(i,j);//指针交换
}
void Operation(int START,int END)
{
    i=START;
    j=END;
    while(i!=j)
    {
        if(i<j)
        {
            if(a[i]>a[j])
                Swap();//指针互换
            else
                j--;
        }
        else
        {
            if(a[i]<a[j])
                Swap();
            else
                j++;
        }
    }
    if(i<k)
    {
        Operation(i+1,END);
    }
    else if(i==k)
    {
        for(l=1; l<=m; l++)
        {
            if(b[l]==a[i])//找到第k小数在原来数组中的位置
            {
                cout<<l<<' ';
                break;
            }
        }
    }
    else
    {
        Operation(START,i-1);
    }
}
int main()
{
    while(scanf("%d%d",&m,&k)!=EOF)
    {
        for(i=1; i<=m; i++)
        {
            cin>>a[i];
            b[i]=a[i];
        }
        for(i=6; i<=m+1; i+=5)
        {
            stable_sort(a+i-5,a+i,cmp);//将数据分为5个一组，进行组内排序
        }
        for(i=1; i<=m/5; i++)
        {
            swap(a[i],a[i*5-2]);//将每组数据中的中位数交换到整个数组的第一位
        }
        stable_sort(a+1,a+i,cmp);
        swap(a[1],a[i/2]);//将中位数的中位数放到数组的第一位
        Operation(l,m);
    }
    return 0;
}

就此我们可以改写一下此程序，求第K大数所在位置，即求第m-k+1小的数在数组中的位置

代码如下

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int a[100005],b[100005];
int i,j,m,l,k;
int cmp(int a,int b)
{
    return a<b;
}
void Swap()
{
    swap(a[i],a[j]);
    swap(i,j);//指针交换
}
void Operation(int START,int END,int ans)
{
    i=START;
    j=END;
    while(i!=j)
    {
        if(i<j)
        {
            if(a[i]>a[j])
                Swap();//指针互换
            else
                j--;
        }
        else
        {
            if(a[i]<a[j])
                Swap();
            else
                j++;
        }
    }
    if(i<ans)
    {
        Operation(i+1,END,ans);
    }
    else if(i==ans)
    {
        for(l=1; l<=m; l++)
        {
            if(b[l]==a[i])//找到第k小数在原来数组中的位置
            {
                cout<<l<<' ';
                break;
            }
        }
    }
    else
    {
        Operation(START,i-1,ans);
    }
}
int main()
{
    while(scanf("%d%d",&m,&k)!=EOF)
    {
        for(i=1; i<=m; i++)
        {
            cin>>a[i];
            b[i]=a[i];
        }
        for(i=6; i<=m+1; i+=5)
        {
            stable_sort(a+i-5,a+i,cmp);//将数据分为5个一组，进行组内排序
        }
        for(i=1; i<=m/5; i++)
        {
            swap(a[i],a[i*5-2]);//将每组数据中的中位数交换到整个数组的第一位
        }
        stable_sort(a+1,a+i,cmp);
        swap(a[1],a[i/2]);//将中位数的中位数放到数组的第一位
        Operation(l,m,m-k+1);
    }
    return 0;
}