hdu 5722 线段树+扫描线
来源:互联网 发布:淘宝店铺童装简介怎么 编辑:程序博客网 时间:2024/05/03 10:26
链接:戳这里
Problem Description
After all the difficulties, Psyche and Cupid are finally getting married.
No ordinary pearls can hold Cupid's love for Psyche. So he has collected the Jewelry of Gods at the top of Mount Olympus to make her a chain.
There are n beads on the chain. The i-th bead is of the type Ai.
Being pretty in appearance and generous in her heart, Psyche decides to give one consecutive part of the chain to her mother.
To meet her mother's particular taste, that part must have at least one type of bead that appears x times exactly.
Psyche wants to know the number of ways to choose a part of the chain for her mother.
Note that two parts [L1,R1] and [L2,R2] are different only if L1≠L2 or R1≠R2.
Input
The first line of the input contains an integer T (1≤T≤15), which denotes the number of test cases.
For each test case, the first line contains two integers n,x (1≤n≤105,1≤x≤n).
The second line contains n integers, the i-th integer denotes Ai (0≤Ai≤109).
Output
For each test case, print an integer which denotes the number of parts Psyche can choose.
Sample Input
2
3 1
1 2 1
4 2
2 3 2 2
Sample Output
6
3
Hint
In the first example, all solutions all valid.
In the second example, solution $ [1,3], [2,4], [3,4] $ have a type of beed, 2, that appears twice exactly.
题意:
给出n个数ai,问存在多少个区间满足区间内至少存在一个数出现恰好x次
思路:
枚举区间的右边界,找出符合条件的左边界(对应一个区间)
n = 10 , x = 2
pos : 1 2 3 4 5 6 7 8 9 10
val : . a . a a . . a a .
i=5 那么满足条件的左区间为[1,2]
i=8 满足条件的左区间为[3,4]
i=9 满足条件的左区间为[5,5]
显然答案取区间并,怎么快速统计呢?
pre[i] 表示当前位置i,上一个 ai 出现的位置
R[i] 表示当前位置为i,已经出现了x个相同的数,R[i]=i-x出现的位置
L[i] 表示当前位置为i,已经出现了x+1个相同的数,L[i]=i-x-1出现的位置
具体的看代码吧
代码:
#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<string>#include<vector>#include <ctime>#include<queue>#include<set>#include<map>#include<list>#include<stack>#include<iomanip>#include<cmath>#include<bitset>#define mst(ss,b) memset((ss),(b),sizeof(ss))///#pragma comment(linker, "/STACK:102400000,102400000")typedef long long ll;typedef long double ld;#define INF (1ll<<60)-1#define Max 1e9using namespace std;int T;int a[100100],b[100100];int n,X;ll sum[500100],lazy[500100];void build(int root,int l,int r){ sum[root]=0; lazy[root]=0; if(l==r) return ; int mid=(l+r)/2; build(root*2,l,mid); build(root*2+1,mid+1,r);}void update(int root,int l,int r,int x,int y,int v){ //printf("l=%d r=%d x=%d y=%d\n",l,r,x,y); if(x==l && y==r) { lazy[root]+=v; if(lazy[root]) sum[root]=(r-l+1); else if(l==r) sum[root]=0; else sum[root]=sum[root*2]+sum[root*2+1]; return ; } int mid=(l+r)/2; if(y<=mid) update(root*2,l,mid,x,y,v); else if(x>mid) update(root*2+1,mid+1,r,x,y,v); else { update(root*2,l,mid,x,mid,v); update(root*2+1,mid+1,r,mid+1,y,v); } if(lazy[root]) sum[root]=r-l+1; else sum[root]=sum[root*2]+sum[root*2+1];}int L[100100],R[100100],pre[100100];vector<int> V[100100];int main(){ scanf("%d",&T); while(T--){ scanf("%d%d",&n,&X); for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[i]=a[i]; sort(b+1,b+n+1); int cnt=unique(b+1,b+n+1)-(b+1); for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+cnt+1,a[i])-b; mst(L,0);mst(R,0);mst(pre,0); for(int i=1;i<=cnt;i++) V[i].clear(); for(int i=1;i<=n;i++){ int x=a[i],num=V[a[i]].size(); if(num>0) pre[i]=V[x][num-1]; V[x].push_back(i); num++; if(num>=X) R[i]=V[x][num-X]; if(num>=X+1) L[i]=V[x][num-X-1]; } //for(int i=1;i<=n;i++) printf("%d %d %d\n",L[i],R[i],pre[i]); build(1,1,n); ll ans=0LL; for(int i=1;i<=n;i++){ if(pre[i] && R[pre[i]]) update(1,1,n,L[pre[i]]+1,R[pre[i]],-1); if(R[i]) update(1,1,n,L[i]+1,R[i],1); ans+=1LL*sum[1]; } printf("%I64d\n",ans); } return 0;}/*104 22 3 2 2*/
- hdu 5722 线段树+扫描线
- hdu 1542 线段树扫描线
- hdu 3265 Posters 线段树+扫描线
- hdu 1828 线段树+扫描线
- hdu 3265 线段树+扫描线
- hdu 4007 扫描线 线段树
- HDU 3265 线段树 扫描线
- hdu 1542 Atlantis 线段树扫描线
- HDU 1542 Atlantis(线段树:扫描线)
- HDU 1828 Picture(线段树:扫描线)
- HDU 3265 Posters(线段树:扫描线)
- HDU 3255 Farming(线段树:扫描线)
- 【HDU】1542 Atlantis 线段树+扫描线
- HDU 1542 Atlantis(线段树扫描线)
- HDU - 1542 (扫描线+线段树优化)
- hdu 1264 线段树+扫描线
- HDU 5091 线段树扫描线
- hdu 5091 线段树+扫描线
- Jquery-Mobile: Append the data to the non-native select option menu
- Android布局优化之过度绘制
- 关于js传值
- 使用String类的compareTo()方法比较两个日期字符串
- 测试mysql的sql语句预编译效果
- hdu 5722 线段树+扫描线
- oracle 11g建立DBLink访问sql server数据库
- ShaderForge简单实例2
- 深入C/C++之基于CheckStackVars的安全检查(VS2008)
- python中的类属性和实例属性
- 实现语音视频录制
- 高级定时器
- 加入跨域支持后ajax请求返回Uncaught SyntaxError: Unexpected token :
- java 8 部分新特性和函数式编程