HH的项链(可持久化线段树解法)

想要知道[L,R][L,R][L,R]区间中不同的数是多少
w[i]w[i]w[i]是区间中第iii个数xxx上一次出现的位置
如果w[i]<Lw[i]<Lw[i]<L,那么答案+1
我们用[L,R][L,R][L,R]这一段的w[i]w[i]w[i]来建立一个权值线段树,然后问询这个权值线段树中[1,L−1][1,L-1][1,L1],就可以得到答案
但问询很多,显然我们不能一次问询建一次线段树。
所以可以用可持久化线段树来实现
同时为了防止出现0,所以w[i]w[i]w[i]均+1
https://ac.nowcoder.com/acm/problem/20325

#pragma GCC optmize(2)
#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 1e6 + 10;
int last[N], w[N], root[N]; //上一个x出现在 第i个数上的数上一次出现在哪
int idx = 0;
int n, m;

struct node {
	int l, r;
	int sum;
} tr[N * 30];

int build(int l, int r) {
	int p = ++idx;
	if (l == r)
		return p;
	int mid = l + r >> 1;
	tr[p].l = build(l, mid), tr[p].r = build(mid + 1, r);
	return p;
}

int insert(int p, int l, int r, int x) {
	int q = ++idx;
	tr[q] = tr[p];
	if (l == r) {
		tr[q].sum++;
		return q;
	}
	int mid = l + r >> 1;
	if (x <= mid) {
		tr[q].l = insert(tr[p].l, l, mid, x);
	} else {
		tr[q].r = insert(tr[p].r, mid + 1, r, x);
	}
	tr[q].sum = tr[tr[q].l].sum + tr[tr[q].r].sum;
	return q;
}

int query(int p, int q, int l, int r, int x) {
	if (l == r) {
		return tr[q].sum - tr[p].sum;
	}
	int mid = l + r >> 1;
	int sum = tr[tr[q].l].sum - tr[tr[p].l].sum;
	if (x <= mid) {
		return query(tr[p].l, tr[q].l, l, mid, x);
	} else {
		return sum + query(tr[p].r, tr[q].r, mid + 1, r, x);
	}
}

void solve() {
	cin >> n;
	for (int i = 1, x; i <= n; i++) {
		cin >> x;
		w[i] = last[x];
		last[x] = i;
	}
	root[0] = build(1, n);
	for (int i = 1; i <= n; i++) {
		root[i] = insert(root[i - 1], 1, n, w[i] + 1);
	}

	cin >> m;
	while (m--) {
		int l, r;
		cin >> l >> r;
		cout << query(root[l - 1], root[r], 1, n, l) << endl;
	}

}

signed main() {
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	solve();
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值