题目大意:有一些珠子串成的项链,珠子有不同的颜色。多次询问一段区间内有多少不同的颜色。
思路:这个题让我学会了一种巧妙的离线做法。将问题按左端点排序。处理出来每个颜色第一个出现的位置,和每个颜色下一个出现的位置。然后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;
}