第八大奇蹟【區間查詢第8大數】

誰是凶手1703發表於2020-11-03

第八大奇蹟

單點修改、區間查詢第8大數

思路

線段樹,節點裡面必須存,區間的第8大的數。

考慮,父節點能不能由倆子節點更新。

直接是不能的,得加一些變數,為了維護父節點的第8大數,我們得知道,子節點的前8個數。

然後像歸併排序一樣,更新。

程式碼

#include<iostream>
#include<queue>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
const int N=1e5+100;
int n,m;
struct node
{
    int l,r;
    int v[10];//從大到小
};
node tr[N*4];

bool cmp(int a,int b)
{
    return a>b;
}
void pushup(node &u,node &l,node &r)
{
    
    int a[10]={0},b[10]={0},c[10]={0};
    for(int i=0;i<8;i++)
    {
        a[i]=l.v[i];
        b[i]=r.v[i];
    }
    // sort(a,a+8,cmp);
    // sort(b,b+8,cmp);
    
    int i=0,j=0;
    int k=0;
    while(i<8 && j<8 && k<8)
    {
        if(a[i]>=b[j]) c[k++]=a[i++];
        else c[k++]=b[j++];
    }
    while(i<8 && k<8) c[k++]=a[i++];
    while(j<8 && k<8) c[k++]=b[j++];
    for(int i=0;i<8;i++)
        u.v[i]=c[i];
}
void pushup(int u)
{
   // cout<<"pushup=="<<u<<endl;
    pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
void build(int u,int l,int r)
{
    if(l==r)
    {
        tr[u]={l,r};
        
    }
    else{
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
    }
}

void modify(int u,int x,int v)
{
   // cout<<"--"<<endl;
    if(tr[u].l==tr[u].r)
    {
       // cout<<u<<endl;
        tr[u].v[0]=v;
        
        // for(int i=0;i<8;i++)
        //     cout<<tr[u].v[i]<<' ';
        // cout<<endl;
        //pushup(u);
    }
    else{
        int mid=tr[u].l+tr[u].r>>1;
        if(x<=mid)
            modify(u<<1,x,v);
        else 
            modify(u<<1|1,x,v);
        pushup(u);
    }
}
node query(int u,int l,int r)
{
   
    if(tr[u].l>=l && tr[u].r <=r)
    {
        //cout<<u<<endl;
        // for(int i=0;i<8;i++)
        //     cout<<tr[u].v[i]<<' ';
        // cout<<endl;
        return tr[u];
    }
    else{
        int mid=tr[u].l+tr[u].r>>1;
        if(r<=mid) return query(u<<1,l,r);
        else if(l>mid) return query(u<<1|1,l,r);
        else{
            node left,right,fin;
            left=query(u<<1,l,r);
            right=query(u<<1|1,l,r);
            pushup(fin,left,right);
            return fin;
        }
        
    }
}
int main()
{
   
    scanf("%d%d",&n,&m);
    build(1,1,n);
  
    while(m--){
        char op[2];
        int a,b;
        //cin>>op>>a>>b;
        scanf("%s%d%d",op,&a,&b);
        if(op[0]=='C')
        {
           
            modify(1,a,b);
            
            
        }
        else {
            printf("%d\n",query(1,a,b).v[7]);
            
        }
    }
    
   
    
    return 0;
}

相關文章