BZOJ 1878 SDOI 2009 HH的项链 树状数组 + 离线处理

题目大意:有一些珠子串成的项链,珠子有不同的颜色。多次询问一段区间内有多少不同的颜色。


思路:这个题让我学会了一种巧妙的离线做法。将问题按左端点排序。处理出来每个颜色第一个出现的位置,和每个颜色下一个出现的位置。然后1到cnt循环,如果这里有一个问题的左端点是当前节点,就处理他的答案,方法是前缀合,可以用树状数组。然后把这个颜色的下一个出现的位置+1。

这样做就避免了一种颜色在询问中被处理两次。


CODE:


#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAX 1000010
using namespace std;

struct Complex{
    int x,y,_id;

    bool operator <(const Complex &a)const {
        return x < a.x;
    }
    void Read() {
        scanf("%d%d",&x,&y);
    }
}ask[MAX];

int cnt,src[MAX],asks;
int ans[MAX];
int tree[MAX];
int _next[MAX],last_color[MAX];

void Fix(int x);
int GetSum(int x);

int main()
{
    cin >> cnt;
    for(int i = 1;i <= cnt; ++i)
        scanf("%d",&src[i]);
    cin >> asks;
    for(int i = 1;i <= asks; ++i) {
    	ask[i].Read();
    	ask[i]._id = i;
    }
    sort(ask + 1,ask + asks + 1);
    for(int i = 1;i <= cnt; ++i) {
    	if(!last_color[src[i]])	Fix(i);
    	_next[last_color[src[i]]] = i;
    	last_color[src[i]] = i;
    }
    int p_ask = 1;
    for(int i = 1;i <= cnt; ++i) {
        while(ask[p_ask].x == i) {
    		ans[ask[p_ask]._id] = GetSum(ask[p_ask].y) - GetSum(ask[p_ask].x - 1);
            p_ask++;
        }
        if(_next[i])
            Fix(_next[i]);
    }
    for(int i = 1;i <= asks; ++i)
    	printf("%d\n",ans[i]);
    return 0;
}  

void Fix(int x)
{   
    for(int i = x;i <= cnt;i += i&-i)
        tree[i]++;
}

int GetSum(int x)
{  
    int re = 0;
    for(int i = x;i > 0;i -= i&-i)
        re += tree[i];
    return re;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值