线段树

最新推荐文章于 2022-04-04 16:59:13 发布

sooowhat

最新推荐文章于 2022-04-04 16:59:13 发布

阅读量260

点赞数

分类专栏：暑期总结 poj 文章标签： insert query c struct

本文链接：https://blog.csdn.net/sooowhat/article/details/6677751

版权

暑期总结同时被 2 个专栏收录

7 篇文章 0 订阅

订阅专栏

poj

7 篇文章 0 订阅

订阅专栏

//21:38 started
#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
int L,T,O;
#define N   100005
struct Node
{
    int l,r;
    int mid;
    int color;
};
Node node[6*N];
int mark[40];
int cnt;
void init(int id,int l,int r);
void query(int id,int l,int r);
void insert(int id,int l,int r,int c);
int main()
{
    freopen("in.txt","r",stdin);
    char flag;
    cin>>L>>T>>O;
    init(1,0,L);

    int x,y,c;
    for(int i=0;i<O;i++)
    {
        cin>>flag;
        if(flag=='C')
        {
            scanf("%d%d%d",&x,&y,&c);
            insert(1,x-1,y,c);
    //   cout<<x<<" "<<y<<endl;
        }
        else if(flag=='P')
        {
            cnt=0;
            memset(mark,0,sizeof(mark));
           scanf("%d%d",&x,&y);
            if(x>y) {
            int tmp=x;
            x=y; y=x;
            }
             query(1,x-1,y);
            printf("%d\n",cnt);
        }

    }
}

void init(int id,int l,int r)
{
    node[id].l=l;node[id].r=r;
    node[id].mid=(l+r)/2;
    node[id].color=1;
    if(l+1==r) return ;
    init(2*id,l,node[id].mid);
    init(2*id+1,node[id].mid,r);
}

void insert(int id,int l,int r,int c)
{
      if(node[id].l==l&&node[id].r==r){
         node[id].color=c; return ;
      }
     if(node[id].color!=0) {
         node[2*id].color=node[id].color;
         node[2*id+1].color=node[id].color;
         node[id].color=0;
         }


      if(r<=node[id].mid)
      insert(2*id,l,r,c);
      else if(l>=node[id].mid)
      insert(2*id+1,l,r,c);
      else
      {
          insert(2*id,l,node[id].mid,c);
          insert(2*id+1,node[id].mid,r,c);
      }

}

void query(int id,int l,int r)
{
     if(node[id].color!=0)
     {
         if(!mark[node[id].color]) {mark[node[id].color]=1; cnt++;}
         return ;
     }

    else {

        if(l>=node[id].mid) query(2*id+1,l,r);
        else if(r<=node[id].mid) query(2*id,l,r);
        else {
     query(2*id,l,node[id].mid);
     query(2*id+1,node[id].mid,r);
        }
    }
}

pKu 2777 写的比较干净 ,TLE了很多次应该用scanf的

pku2828 和 2886 让我对线段树的用法有了进一步的理解, 线段树可以像二叉查找树一样, 左右都有一个值代表的是这个范围内的和或者这个范围内的最大..等等,可以动态的修改这个值,使左边或右边的和减小. 也就是实现动态插入的功能, 比如约瑟夫环的问题.