题解 | #牛牛的数列#

Segment Tree

https://ac.nowcoder.com/acm/problem/13892

链接:https://ac.nowcoder.com/acm/problem/13892

今天HH正在学习一种新的数据结构,叫做segment tree,经常用来解决segment问题,这里来一个: 给你一个长度为 n,(a 1,a 2 ,…, a n )的无序序列,现在你应该计算有多少段 L,R 满足两个条件: 1.段的长度为k(即R-L+1=k)。 2.L和R(都包括)之间的数字总共至少出现q次。 HH 认为这个问题太简单了,所以他把问题交给了你。

这道题目可以使用桶排加上前缀和的思路来解决(刚开始还以为是尺取法.....) 前缀和累加前n项的和,减去k项之前的前缀和就求出了区间内数字出现的总个数

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
int a[200];
int tot[400];
int sum[400];
int main() {
	int T;cin>>T;
	while(T--) {
		int n,k,q;cin>>n>>k>>q;
		int maxx=-999;
		memset(tot,0,sizeof(tot));
		memset(a,0,sizeof(a));
		memset(sum,0,sizeof(sum));
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			tot[a[i]]++;
			maxx=max(a[i],maxx);
		}
		sum[0]=0;
		int res=0;
		sum[1]=tot[1];
		for(int i=2;i<=maxx+k-1;i++) {
			sum[i]=sum[i-1]+tot[i];
		}
		for(int i=k;i<=maxx+k-1;i++) {
			if(sum[i]-sum[i-k]>=q) {
				res+=1;
			}
		}
		printf("%d\n",res);
	}
	return 0;
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务