【洛谷P1972】HH的项链(求区间内有多少个不同的数---树状数组)

it2022-05-05  114

题目地址:https://www.luogu.org/problemnew/show/P1972

题目:


题目描述

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。

输入输出格式

输入格式:

第一行:一个整数N,表示项链的长度。

第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。

第三行:一个整数M,表示HH 询问的个数。

接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。

输出格式:

M 行,每行一个整数,依次表示询问对应的答案。

输入输出样例

输入样例#1: 

6 1 2 3 4 3 5 3 1 2 3 5 2 6

输出样例#1: 

2 2 4

说明

数据范围:

对于100%的数据,N <= 500000,M <= 500000。

解题思路:


模版题,离线处理。

网上找到了两种做法,把我死磕了好久的那种方法当作第二种吧。

先对输入的【l,r】以r为关键字从小到大排序,一个虚拟数组cnt[](之所以说它是虚拟数组,因为可以用树状数组实现想要的操作,不需要真的定义这么一个数组)。

--->第一种:

last[i]记录i位置上的数上一次出现的位置,当遍历给定序列的下标i时,cnt[i]+1,如果i位的这个数之前出现过,即存在last[i],则将cnt[last[i]]-1(在遍历到i之前已经遍历过last[i]位置了cnt[last[i]]=1,所以这样操作下来,cnt[last[i]]=0),不会对当前要求的结果造成影响。(请参考代码理解,在区间右边界一定的前提下该解法是正确的)

比如序列:2,3,4,3,4,2

当右边界为5时(下标从1开始),对应的cnt[]值分别为:1,0,0,1,1

所以区间【1,5】内不同的数字个数=3,区间【2,5】内不同的数字个数=2,区间【4,5】内不同的数字个数=2(树状数组getsum求解)

因为右边界一定,所以右边界左侧不统计重复出现的数,每个数只统计一次,确保答案正确

--->第二种:

在第一种的基础上做了巧妙的变化,front[i]记录i位置上的数上一次出现的位置+1。。。有点不是很好理解。。目前我也没有完全理解透彻,暂且掌握第一种吧。

ac代码:


方法1:

#include <bits/stdc++.h> using namespace std; #define maxn 5000010 #define lowbit(x) ((x) & (-x)) int n, m, tmp; int c[maxn],last[maxn],exist[maxn]; int ans[maxn]; struct node { int l,r,no; friend bool operator < (node a, node b) { return a.r == b.r ? a.l < b.l : a.r < b.r; } }a[maxn]; void update(int x,int v) { for(int i = x; i < maxn; i += lowbit(i)) c[i] += v; } int query(int x) { int sum=0; for(int i = x; i > 0; i -= lowbit(i)) sum += c[i]; return sum; } int main() { //freopen("/Users/zhangkanqi/Desktop/11.txt", "r", stdin); scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&tmp); last[i] = exist[tmp];//记录上一个tmp出现的位置 exist[tmp] = i; } scanf("%d",&m); for(int i=1;i<=m;i++) { scanf("%d%d",&a[i].l,&a[i].r); a[i].no=i;//每次查询的序号 } sort(a+1,a+1+m);//按右端点排序 int k=0; for(int i = 1; i <= m; i++) { while(k < a[i].r) { k++; if(last[k]) update(last[k], -1);//把之前的值调0,1+(-1) update(k, 1); } ans[a[i].no] = query(a[i].r) - query(a[i].l - 1); } for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }

方法2:

#include <bits/stdc++.h> using namespace std; #define maxn 5000010 #define lowbit(x) ((x) & (-x)) int n,m; int c[maxn],front[maxn],last[maxn]; int ans[maxn]; struct node { int l,r,no; friend bool operator < (node a, node b) { return a.r == b.r ? a.l < b.l : a.r < b.r; } }a[maxn]; void update(int x,int v) { for(int i = x; i < maxn; i += lowbit(i)) c[i] += v; } int query(int x) { int sum=0; for(int i = x; i > 0; i -= lowbit(i)) sum += c[i]; return sum; } int main() { //freopen("/Users/zhangkanqi/Desktop/11.txt", "r", stdin); scanf("%d",&n); for(int i=1;i<=n;i++) { int temp; scanf("%d",&temp); front[i]=last[temp]+1;//front数组为i位置上的数上一次出现的位置+1 last[temp]=i;//last数组为i位置上的数到目前为止最后一次出现的位置 } scanf("%d",&m); for(int i=1;i<=m;i++) { scanf("%d%d",&a[i].l,&a[i].r); a[i].no=i;//每次查询的序号 } sort(a+1,a+1+m);//按右端点排序 int k=1;//k初值为1 for(int i=1;i<=m;i++) { while(k<=a[i].r) { k++; update(front[k-1],1); update(k,-1); } ans[a[i].no]=query(a[i].l);//查询左端点 } for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }

【参考博客】:

https://blog.csdn.net/xqx1343002589/article/details/81296741

https://blog.csdn.net/sunshiness_s/article/details/80429470

 


最新回复(0)