【HDU6183】Color it(线段树)

最新推荐文章于 2026-06-04 15:39:18 发布

原创最新推荐文章于 2026-06-04 15:39:18 发布 · 1.6k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#线段树 #数据结构 #hdu

------------acm------------ 同时被 3 个专栏收录

117 篇文章

订阅专栏

数据结构

34 篇文章

订阅专栏

树

20 篇文章

订阅专栏

本文介绍了一道涉及四种操作的颜色计数问题，并通过构建51棵线段树来解决该问题。每棵树沿Y轴建立，记录指定范围内最小X值。最终实现了在特定矩形区域内统计不同颜色的数量。

记录一个菜逼的成长。。

题目链接

题目大意:

有四种操作。
$0$ :清除所有点
$1\ x\ y\ c$ :给点 $(x,y)$ 添加一种颜色 $c$
$2\ x\ y1\ y2$ :在 $(0,y1)$ 与 $(x,y2)$ 所围成的矩形里有多少种颜色
$3$ :程序结束

笔记

注意这里一个点可以有很多种颜色，是不会被覆盖的。
颜色最多51种。我们就建51棵线段树。
每个线段树按 $y$ 轴建树，每个结点的值是在范围内的最小的 $x$ 值
ps:看了cls(claris)的cpp感觉学到了姿势啊。orz.
时间复杂度 $O(50nlogn)$

#include <bits/stdc++.h>
using namespace std;
#define rep(i,l,r) for( int i = l; i <= r; i++ )
#define rep0(i,l,r) for( int i = l; i < r; i++ )
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define clr clear()
#define pb push_back
#define mp make_pair
#define fi first
#define se second
const int maxn = 1000000 + 10;
int T[100],l[maxn],r[maxn],v[maxn];
int tot,X,y1,y2,flag;
void ins(int &t,int L,int R,int y,int x) {
  if(!t){
    t = ++tot;
    v[t] = x;
  }
  if(v[t] > x)v[t] = x;//保存最小的x值
  if(L == R)return ;
  int mid = (L + R) >> 1;
  if(y <= mid)ins(l[t],L,mid,y,x);
  else ins(r[t],mid+1,R,y,x);
}
void ask(int t,int L,int R) {
  if(flag || !t)return ;
  if(y1 <= L && y2 >= R){
    if(v[t] <= X)flag = 1;
    return ;
  }
  int mid = (L + R) >> 1;
  if(y1 <= mid)ask(l[t],L,mid);
  if(y2 >  mid)ask(r[t],mid+1,R);
}
int main()
{
  for (int i = 0; i <= 50; i++)T[i] = 0;
  int ope;
  while (~scanf("%d",&ope)) {
    if (ope == 3)return 0;
    if (ope == 0) {
      for (int i = 0; i <= 50; i++)T[i] = 0;
      for (int i = 1; i <= tot; i++) l[i] = r[i] = 0;
      tot = 0;
    }
    else if (ope == 1) {
      int x,y,c;
      scanf("%d%d%d",&x,&y,&c);
      ins(T[c],1,1000000,y,x);
    }
    else if(ope == 2){
      scanf("%d%d%d",&X,&y1,&y2);
      int ans = 0;
      for (int i = 0; i <= 50; i++) {
        flag = 0;
        ask(T[i],1,1000000);
        if(flag)ans++;
      }
      printf("%d\n",ans);
    }
  }
  return 0;
}