HDU2665 求区间第K大 主席树

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=2665

代码：

//#include<bits/stdc++.h>
#include<iostream>
#include<cmath>
#include<algorithm>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
#define lowbit(x) ((x)&(-x))
using namespace std;
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=1e5+;
typedef long long LL;
typedef unsigned long long ULL;
const LL mod=1e9+;
const ULL base=1e7+;
struct node{
    int siz,lc,rc;
}a[maxn*];
int s[maxn];
struct nub{
    int st,num;
    bool operator<(const nub &u)const{
        return num<u.num;
    }
}u[maxn];
int tot=,rt[maxn];
void Build(int d,int l,int r){
    a[d].siz=;
    if(l==r){
        return ;
    }
    int m=(l+r)>>;
    a[d].lc=tot++;a[d].rc=tot++;
    Build(a[d].lc,l,m);
    Build(a[d].rc,m+,r);
}
void updata(int st,int d,int last,int l,int r){
    a[d].siz=a[last].siz+;
    if(l==r){
        return ;
    }
    int mid=(l+r)>>;
    if(st<=mid){
        a[d].rc=a[last].rc;
        a[d].lc=tot++;
        updata(st,a[d].lc,a[last].lc,l,mid);
    }
    else if(st>mid){
        a[d].lc=a[last].lc;
        a[d].rc=tot++;
        updata(st,a[d].rc,a[last].rc,mid+,r);
    }
}
int query(int k,int last,int now,int l,int r){
    if(l==r){
        return l;
    }
    int mid=(l+r)>>;
    if(k<=a[a[now].lc].siz-a[a[last].lc].siz){
        return query(k,a[last].lc,a[now].lc,l,mid);
    }
    else{
        return query(k-(a[a[now].lc].siz-a[a[last].lc].siz),a[last].rc,a[now].rc,mid+,r);
    }
}
int b[maxn];
int main(){
    int n,m,t;
    scanf("%d",&t);
    while(t--){
        tot=;
        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++){
            scanf("%d",&u[i].num);
            u[i].st=i;
        }
        sort(u+,u++n);
        int cnt=;
        s[cnt]=u[].num;
        b[u[].st]=;
        for(int i=;i<=n;i++){
            if(u[i].num!=u[i-].num){
                cnt++;
                s[cnt]=u[i].num;
            }
            b[u[i].st]=cnt;
        }
        rt[]=tot++;
        Build(rt[],,cnt);
        for(int i=;i<=n;i++){
            rt[i]=tot++;
            updata(b[i],rt[i],rt[i-],,cnt);
        }
        int l,r,k;
        for(int i=;i<=m;i++){
            scanf("%d%d%d",&l,&r,&k);
            printf("%d\n",s[query(k,rt[l-],rt[r],,cnt)]);
        }
    }
    system("pause");
    return ;
}

HDU1007

http://acm.hdu.edu.cn/showproblem.php?pid=1007

方差优化：

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
#define lowbit(x) ((x)&(-x))
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=2e5+;
typedef long long LL;
typedef unsigned long long ULL;
const LL mod=1e9+;
const ULL base=1e7+;
using namespace std;
int kdd;
struct node{
    int id,di;
    double g[];
    bool operator<(const node &u)const{
       return g[kdd]<u.g[kdd];
    }
}kdt[maxn<<],data[maxn];
bool flag[maxn<<];
double dis(node a,node b){
    return (a.g[]-b.g[])*(a.g[]-b.g[])+(a.g[]-b.g[])*(a.g[]-b.g[]);
}
double fx(int l,int r,int j){
    double as=0.0;
    for(int i=l;i<=r;i++){
        as+=data[i].g[j];
    }
    as/=(double)(r-l+);
    double f=0.0;
    for(int i=l;i<=r;i++){
        f+=(data[i].g[j]-as)*(data[i].g[j]-as);
    }
    f/=(double)(r-l+);
    return f;
}
void build(int l,int r,int rt){
    if(l>r) return ;
    if(fx(l,r,)>fx(l,r,)){
        kdd=;
    }
    else{
        kdd=;
    }
    flag[rt]=;
    int mid=(l+r)>>;
    nth_element(data+l,data+mid,data+r+);
    kdt[rt]=data[mid];
    kdt[rt].di=kdd;
    flag[rt<<]=flag[rt<<|]=;
    if(l<=mid-){
        build(l,mid,rt<<);
    }
    if(mid+<=r){
        build(mid+,r,rt<<|);
    }
}
pair<double,node> ans;
double mi;
void query(int rt,node p){
    int kd=kdt[rt].di;
    pair<double,node> now={dis(p,kdt[rt]),kdt[rt]};
    int x=rt<<;
    int y=rt<<|;
    if(p.g[kd]>=kdt[rt].g[kd]){
        swap(x,y);
    }
    if(flag[x])
        query(x,p);
    bool ff=;
    if(ans.fi==-1.0){
        ff=;
        if(p.id!=kdt[rt].id){
            ans=now;
        }
    }
    else{
        if(p.id!=kdt[rt].id&&(now.fi<ans.fi||now.fi==ans.fi&&now.se.id<ans.se.id)){
            ans=now;
        }
        if((kdt[rt].g[kd]-p.g[kd])*(kdt[rt].g[kd]-p.g[kd])<ans.fi){
            ff=;
        }
    }
    if(flag[y]&&ff){
        query(y,p);
    }
}
int main(){
    int n;
    while(~scanf("%d",&n)&&n!=){
        for(int i=;i<=n;i++){
            for(int j=;j<;j++){
                scanf("%lf",&data[i].g[j]);
            }
            data[i].id=i;
        }
        build(,n,);
        mi=1000000000.0;
        for(int i=;i<=n;i++){
            ans.fi=-1.0;
            query(,data[i]);
            mi=min(mi,sqrt(ans.fi));
        }
        mi/=2.0;
        printf("%.2lf\n",mi);
    }
}