首页 > 代码库 > BZOJ 1878 SDOI 2009 HH的项链 树状数组 + 离线处理
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; }
BZOJ 1878 SDOI 2009 HH的项链 树状数组 + 离线处理
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。