2019.10.02题解

T1

一般都能想到二分+取前m大正值，但是复杂度无法承受，我们发现要的是sum值，并不需要每个位置的准确值，

所以用可以nth_element把大于第m大的放右边即可。（原来nth还可以这么用）。

nth_element实现：

每次找一个base，小于base的放右边，大于的放右边

#include<bits/stdc++.h>
using namespace std;
const int N=1e8+10;
int n,k,a[N];
inline int read()
{
        register int sum,k=1;register char s;
        while(s=getchar(),s<'0'||s>'9') if(s=='-') k=-1; sum=s-'0';
        while(s=getchar(),s>='0'&&s<='9') sum=sum*10+s-'0';
        return k*sum;
}
void NTH(int k,int l,int r)
{
        vector<int>v[3];
        int x=a[l],tot=l-1;
        for(int i=l;i<=r;i++) 
        {
                if(a[i]<x) v[0].push_back(a[i]);
                else if(a[i]==x) v[2].push_back(a[i]);
                else v[1].push_back(a[i]);
        }
        for(int i=0;i<v[0].size();i++) a[++tot]=v[0][i];
        for(int i=0;i<v[2].size();i++) a[++tot]=v[2][i];
        for(int i=0;i<v[1].size();i++) a[++tot]=v[1][i];
        if(k<=v[0].size()) NTH(k,l,l+v[0].size()-1);
        else if(k>v[0].size()&&k<=v[0].size()+v[2].size()) return;
        else NTH(k-v[0].size()-v[2].size(),l+v[0].size()+v[2].size(),r);
}
int main()
{
        //freopen("1.in","r",stdin);
        //freopen("1.out","w",stdout);
        n=read(),k=read();
        for(int i=1;i<=n;i++) a[i]=read();
        NTH(k,1,n);
        printf("%d",a[k]);
        return 0;
}

时间复杂度最坏O(n^2)，但一般是O(n)的，空间O(n)(好像可以O(1)但咕了)，跑1e7的点跑800ms。

然而T1卡手打nth_element，我和kx赛后都被卡了。

T2,T3

暂时咕了