poj 2777 Count Color

        到目前为止，poj刷了60多道，之前都做完就算了，想想记录下来也是有好处的，以后可以回过头来看看学习笔记。今天是第一篇博客，刚开始写不好，不过我会学习的。

这是一道线段树的题目，对初学者来说是一道不错的题，直接贴代码了。

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;


struct Node
{
int left,right,mid,color;
}node[400009];


bool color[32];


void makeTree(int l,int r,int p)
{
node[p].left=l;
node[p].right=r;
node[p].mid=(l+r)/2;
node[p].color=1;
if(l==r) return;
makeTree(l,node[p].mid,p<<1);
makeTree(node[p].mid+1,r,p<<1|1);
}


void update(int l,int r,int c,int p)
{
if(node[p].left==l&&node[p].right==r)
{
node[p].color=c;
return;
}
if(node[p].color==c) return;
if(node[p].color!=-1)
{
node[p<<1].color=node[p].color;
node[p<<1|1].color=node[p].color;
node[p].color=-1;
}
if(r<=node[p].mid) update(l,r,c,p<<1);
else if(l>node[p].mid) update(l,r,c,p<<1|1);
else 
{
update(l,node[p].mid,c,p<<1);
update(node[p].mid+1,r,c,p<<1|1);
}
}


void query(int l,int r,int p)
{
if(node[p].color!=-1)
{
color[node[p].color]=true;
return;
}
if(r<=node[p].mid) query(l,r,p<<1);
else if(l>node[p].mid) query(l,r,p<<1|1);
else 
{
query(l,node[p].mid,p<<1);
query(node[p].mid+1,r,p<<1|1);
}
}


int main()
{
int L,T,O;
while(cin>>L>>T>>O)
{
makeTree(1,L,1);
int a,b,c;
char op[2];
while(O--)
{
scanf("%s%d%d",op,&a,&b);
// cin>>op>>a>>b;
if(op[0]=='C')
{
scanf("%d",&c);
// cin>>c;
update(a,b,c,1);
}
if(op[0]=='P')
{
memset(color,false,sizeof(color));
query(a,b,1);

int count=0;
for(int i=0;i<32;i++)
{
if(color[i]) count++;
}
cout<<count<<endl;
}
}
}
}