标签:bzoj1878 问题 algorithm tail int fine oid stdin ret
[BZOJ1878][SDOI2009]HH的项链
试题描述
输入
输出
M行,每行一个整数,依次表示询问对应的答案。
输入示例
6 1 2 3 4 3 5 3 1 2 3 5 2 6
输出示例
2 2 4
数据规模及约定
对于20%的数据,N ≤ 100,M ≤ 1000;
对于40%的数据,N ≤ 3000,M ≤ 200000;
对于100%的数据,N ≤ 50000,M ≤ 200000。
题解
找到每个颜色的前驱(即上一次出现在哪)并维护下来,那么询问时对于一个区间 [L, R],若某个颜色为 c 的点的前驱小于 L,那么它是最靠前的颜色为 c 的点,显然,统计出有多少个“最靠前的某颜色的点的个数”,就能得到答案了。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <cstring>
#include <string>
#include <map>
#include <set>
using namespace std;
const int BufferSize = 1 << 16;
char buffer[BufferSize], *Head, *Tail;
inline char Getchar() {
if(Head == Tail) {
int l = fread(buffer, 1, BufferSize, stdin);
Tail = (Head = buffer) + l;
}
return *Head++;
}
int read() {
int x = 0, f = 1; char c = Getchar();
while(!isdigit(c)){ if(c == ‘-‘) f = -1; c = Getchar(); }
while(isdigit(c)){ x = x * 10 + c - ‘0‘; c = Getchar(); }
return x * f;
}
#define maxn 50010
#define maxnode 1600010
#define maxc 1000010
int ToT, rt[maxn], sumv[maxnode], lc[maxnode], rc[maxnode];
void update(int& y, int x, int l, int r, int p) {
sumv[y = ++ToT] = sumv[x] + 1;
if(l == r) return ;
int mid = l + r >> 1; lc[y] = lc[x]; rc[y] = rc[x];
if(p <= mid) update(lc[y], lc[x], l, mid, p);
else update(rc[y], rc[x], mid + 1, r, p);
return ;
}
int query(int o, int l, int r, int qr) {
if(!o) return 0;
if(r <= qr) return sumv[o];
int mid = l + r >> 1, ans = query(lc[o], l, mid, qr);
if(qr > mid) ans += query(rc[o], mid + 1, r, qr);
return ans;
}
int lstp[maxc];
int main() {
int n = read();
for(int i = 1; i <= n; i++) {
int col = read();
update(rt[i], rt[i-1], 0, n, lstp[col]);
lstp[col] = i;
}
int q = read();
while(q--) {
int ql = read(), qr = read();
printf("%d\n", query(rt[qr], 0, n, ql - 1) - query(rt[ql-1], 0, n, ql - 1));
}
return 0;
}
标签:bzoj1878 问题 algorithm tail int fine oid stdin ret
原文地址:http://www.cnblogs.com/xiao-ju-ruo-xjr/p/6358108.html