poj 2777 Count Color 基础线段树,带给的初学者噩梦!!!

来源:互联网 发布:淘宝supreme正品吧 编辑:程序博客网 时间:2024/05/01 06:07

线段树的精髓是父亲结果要优于孩子结果啊!!!!!用过的线段直接赋值-1; 

#include<iostream>
#include<string.h>
using namespace std;
#define N 100005
struct node
{
 int l,r,c;
}a[N*4];
int flag[35];
void build(int left,int right,int n)
{
 a[n].l=left;
 a[n].r=right;
 a[n].c=1;
 if(a[n].l==a[n].r)
 return ;
 int mid=(left+right)>>1;
 build(left,mid,n*2);
 build(mid+1,right,n*2+1);
}
void updata(int left,int right,int n,int t)
{
 if(a[n].l==left&&a[n].r==right)
 {
  a[n].c=t;
  return ;
 }
 if(a[n].c>=1)//这个地方是核心啊!!
 {
  a[n*2].c=a[n].c;
  a[n*2+1].c=a[n].c;
  a[n].c=-1;
 }
 int mid=(a[n].l+a[n].r)>>1;
 if(mid>=right)
 updata(left,right,n*2,t);
 else if(mid<left)
 updata(left,right,n*2+1,t);
 else
 {
  updata(left,mid,n*2,t);
  updata(mid+1,right,n*2+1,t);
 }
}
void query (int left,int right,int n)
{
 if(a[n].c!=-1)
 {
  flag[a[n].c]=1;
     return ;
 }
 int mid=(a[n].l+a[n].r)>>1;
 if(mid>=right)
 query(left,right,n*2);
 else if(mid<left)
 query(left,right,n*2+1);
 else
 {
  query(left,mid,n*2);
  query(mid+1,right,n*2+1);
 }
}
int main()
{
    int n, t, m, l, r, c, ans;
    char s[10];
    while(scanf("%d%d%d", &n, &t, &m) != EOF)
    {
        build(1,n,1);
        while(m --)
        {
            scanf("%s", s);
            if(s[0] == 'C')
            {
                scanf("%d%d%d", &l, &r, &c); //system("pause");
                updata(l,r,1,c);
              
            }
            else
            {
                scanf("%d%d", &l, &r);
                memset(flag, 0, sizeof(flag));
                query(l,r,1); ans = 0;
                for(int i = 1; i <= 30; i ++) if(flag[i]) ans ++;
                printf("%d\n", ans);
            }
        }
    }
    return 0;

 

原创粉丝点击