线段树查找最大值以及查找区间和

I Hate It

 

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 94892    Accepted Submission(s): 35915

 

 

Problem Description

很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。

 

 

Input

本题目包含多组测试,请处理到文件结束。
在每个测试的第一行,有两个正整数 N 和 M ( 0 学生ID编号分别从1编到N。
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取'Q'或'U') ,和两个正整数A,B。
当C为'Q'的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。
当C为'U'的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。

 

 

Output

对于每一次询问操作,在一行里面输出最高成绩。

 

 

Sample Input

 

5 61 2 3 4 5Q 1 5U 3 6Q 3 4Q 4 5U 2 9Q 1 5

 

Sample Output

 

565

 

查找最大值代码:

#include
#include
# define MAXN 2000005
int maxx;
using namespace std;
struct node
{
    int left;
    int right;
    int value;
} q[MAXN];
int father[MAXN];
void BuildTree(int i,int l,int r)
{
    q[i].left=l;
    q[i].right=r;
    q[i].value=0;
    if(l==r)
    {
        father[l]=i;
        return ;
    }
    BuildTree(i<<1,l,(l+r)/2);
    BuildTree((i<<1)+1,(l+r)/2+1,r);
}
void UpdateTree(int ri)
{
    if(ri==1)
    {
        return ;
    }
    int fi=ri/2;
    int a=q[fi*2].value;
    int b=q[fi*2+1].value;
    q[ri/2].value=max(a,b);
    UpdateTree(ri/2);
}
void FindMax(int i,int l,int r)
{
    if(q[i].left==l&&q[i].right==r)
    {
        maxx=max(maxx,q[i].value);
        return ;
    }
    i=i<<1;
    if(l<=q[i].right)
    {
        if(r<=q[i].right)
        {
            FindMax(i,l,r);
        }
        else if(r>q[i].right)
        {
            FindMax(i,l,q[i].right);
        }
    }
    i++;
    if(r>=q[i].left)
    {
        if(l>=q[i].left)
        {
            FindMax(i,l,r);
        }
        else
            FindMax(i,q[i].left,r);
    }
}

int main()
{
    int n,m;
    ios::sync_with_stdio(false);
    while(cin>>n>>m)
    {
        BuildTree(1,1,n);
        for(int i=1; i<=n; i++)
        {
            int temp;
            cin>>temp;
            q[father[i]].value=temp;
            UpdateTree(father[i]);
        }
        while(m--)
        {
            char str;
            int u,v;
            cin>>str>>u>>v;
            if(str=='Q')
            {
                maxx=0;
                FindMax(1,u,v);
                cout<             }
            else
            {
                q[father[u]].value=v;
                UpdateTree(father[u]);
            }
        }
    }
    return 0;
}
 

查找区间和代码:

#include
#include
# define MAXN 2000005
int maxx;
int res;
using namespace std;
struct node
{
    int left;
    int right;
    int value;
} q[MAXN];
int father[MAXN];
void BuildTree(int i,int l,int r)
{
    q[i].left=l;
    q[i].right=r;
    q[i].value=0;
    if(l==r)
    {
        father[l]=i;
        return ;
    }
    BuildTree(i<<1,l,(l+r)/2);
    BuildTree((i<<1)+1,(l+r)/2+1,r);
}
void UpdateTree(int ri)
{
    if(ri==1)
    {
        return ;
    }
    int fi=ri/2;
    int a=q[fi*2].value;
    int b=q[fi*2+1].value;
    q[ri/2].value=a+b;
    UpdateTree(ri/2);
}
void Query(int i,int l,int r)
{
    if(q[i].left==l&&q[i].right==r)
    {
        res+=q[i].value;
       // maxx=max(maxx,q[i].value);
        return ;
    }
    i=i<<1;
    if(l<=q[i].right)
    {
        if(r<=q[i].right)
        {
            Query(i,l,r);
        }
        else
        {
            Query(i,l,q[i].right);
        }
    }
    i++;
    if(r>=q[i].left)
    {
        if(l>=q[i].left)
        {
            Query(i,l,r);
        }
        else
            Query(i,q[i].left,r);
    }
}

int main()
{
    int n,m;
    ios::sync_with_stdio(false);
    while(cin>>n>>m)
    {
        BuildTree(1,1,n);
        for(int i=1; i<=n; i++)
        {
            int temp;
            cin>>temp;
            q[father[i]].value=temp;
            UpdateTree(father[i]);
        }
        while(m--)
        {
            res=0;
            char str[100];
            int u,v;
            cin>>str>>u>>v;
            if(str[0]=='Q')
            {
                Query(1,u,v);
                cout<             }
            else
            {
                q[father[u]].value+=v;
                UpdateTree(father[u]);
            }
        }
    }
    return 0;
}

 

你可能感兴趣的:(线段树)