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

标签：bzoj   bzoj1878   树状数组   离线处理   

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

思路：这个题让我学会了一种巧妙的离线做法。将问题按左端点排序。处理出来每个颜色第一个出现的位置，和每个颜色下一个出现的位置。然后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;
}

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

标签：bzoj   bzoj1878   树状数组   离线处理   

原文地址：http://blog.csdn.net/jiangyuze831/article/details/39717863