两个区间第K大问题

一:静态无修改

    经典的方法就是使用划分树。。不过划分树写起来有点纠结。。。(时代的眼泪啊)

    现在这个问题一般都是使用主席树来解决,其思想是这样的:对于原序列每一个前缀,都在其上建立一颗以值为轴的线段树(当然要离散化),保存区间内的值出现的次数,当查询一段区间最小值的时候,相当于在 右端点的线段树减去左端点的线段树 所得到的线段树上找第k个有值的点。。。(注意:权值线段树是可做差的,其差线段树的每个节点值相等于两颗原线段树节点值的差。。。所以并不用保存所得线段树)

    但是如果直接开的话。。。空间复杂度4*N*N...太大了。。。

    考虑到每一个线段树比上一个线段树有不同的节点只有 当前加进来的值所在的区间节点。。。一共只有logn个。。所以除了这些节点。。。其他节点都用上一颗线段树的就好了。。。。于是传统的线段树数组建树的方法就不好用了。。。就用指针来建

#include<cstdlib>
#include<cmath>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
int n,m;
int a[100000+10];//原数组
int b[100000+10];//排序离散数组 
int N;
struct node
{
	int sum;
	node *l,*r;
}tree[100000*24+10],*root[100000+10];
node *newnode()
{
	tree[N].l=tree[N].r=NULL;
	tree[N].sum=0;
	return &tree[N++];
}
node *newnode(node *x)
{
	tree[N]=*x;
	return &tree[N++];
}
node *build(int l,int r)
{
	node *x=newnode();
	if(l==r)return x;
	else
	{
		int mid=l+r>>1;
		x->l=build(l,mid);
		x->r=build(mid+1,r);
		x->sum=x->l->sum+x->r->sum;
		return x;
	}
}
node *update(node *last,int tl,int tr,int pos,int val)
{
	node *x=newnode(last);
	if(tl==tr)
	{
		x->sum+=val;
		return x;
	}
	else
	{
		int mid=tl+tr>>1;
		if(pos<=mid)x->l=update(last->l,tl,mid,pos,val);
		else x->r=update(last->r,mid+1,tr,pos,val);
		x->sum=x->l->sum+x->r->sum;
		return x;
	}
}
int query(node *n1,node *n2,int tl,int tr,int k)
{
	if(tl==tr)return tl;
	else
	{
		int mid=tl+tr>>1;
		int lsum=n1->l->sum-n2->l->sum;
		if(lsum>=k)return query(n1->l,n2->l,tl,mid,k);
		else return query(n1->r,n2->r,mid+1,tr,k-lsum);
	}
}
int inti()
{
	memcpy(b,a,sizeof(a));
	sort(b+1,b+n+1);
	int nt=unique(b+1,b+n+1)-(b+1);
	for(int i=1;i<=n;i++)
	{
		int l=1,r=nt;
		while(l<r)
		{
			int mid=l+r>>1;
			if(a[i]<=b[mid])r=mid;
			else l=mid+1;
		}
		a[i]=l;
	}
	return nt;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	int tn=inti();//离散并得到离散后元素个数
	N=0;
	root[0]=build(1,tn);
	for(int i=1;i<=n;i++)root[i]=update(root[i-1],1,tn,a[i],1);
	for(int i=1;i<=m;i++)
	{
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",b[query(root[r],root[l-1],1,tn,k)]);
	}
	return 0;
}
二.动态点更新的区间第K大问题

    这个嘛其实也可以用主席树解决。。。但是我还没学会。

    今天早上学了下线段树套平衡树的方法:现在原序列上建一颗线段树,对每个线段树节点建立一颗平衡树来储存这个区间内所有的点。。。

    更新时,把每个包含当前点的区间(logn个)节点上的平衡树中删掉更新前的值,加入更新后的值。。。

    然后查询是要这样:二分答案的值。。。然后查询此区间包含的每个整区间中小于等于二分值的个数。。。用这个值来与k比较,然后缩小答案范围。。。最后出解

    我平衡树写的是splay。。。因为目前我只会这个,但是速度很不理想,每个点都要超时限3秒多。。。

    以后再来改进下

#include<cstdlib>
#include<cmath>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=100000;
int s[MAXN+10];
struct Node
{
	int sz,val;
	Node *fa,*c[2];
};
struct SPLAY
{
	Node *nil,*root;
	Node *newnode(int val,Node *fa)
	{
		Node *x=new Node;
		x->fa=fa;
		x->val=val;
		x->sz=1;
		x->c[0]=x->c[1]=nil;
		return x;
	}
	void inti(int val)
	{
		nil=newnode(0,nil);//val,fa
		nil->fa=nil->c[0]=nil->c[1]=nil;
		nil->sz=nil->val=0;
		root=newnode(val,nil);
	}
	void up(Node *x)
	{
		x->sz=x->c[0]->sz+x->c[1]->sz+1;
	}
	void rotate(Node *x,int d)
	{
		Node *y=x->fa;
		if(y->fa!=nil)
		{
			if(y->fa->c[0]==y)y->fa->c[0]=x;
			else y->fa->c[1]==x;
		}
		x->fa=y->fa;
		y->fa=x;
		if(x->c[d^1]!=nil)x->c[d^1]->fa=y;
		y->c[d]=x->c[d^1];
		x->c[d^1]=y;
		up(y);
	}
	void splay(Node *x,Node *fa)
	{
		while(x->fa!=fa)
		{
			if(x->fa->fa==fa)
			{
				if(x->fa->c[0]==x)rotate(x,0);//右旋
				else rotate(x,1);//左旋 
			}
			else
			{
				Node *y=x->fa;
				if(y->fa->c[0]==y)
				{
					if(x->fa->c[0]==x)
					{
						rotate(y,0);
						rotate(x,0);
					}
					else
					{
						rotate(x,1);
						rotate(x,0);
					}
				}
				else
				{
					if(x->fa->c[0]==x)
					{
						rotate(x,0);
						rotate(x,1);
					}
					else
					{
						rotate(y,1);
						rotate(x,1);
					}
				}
			}
		}
		up(x);
		if(x->fa==nil)root=x;
	}
	void select(int k,Node *fa)
	{
		Node *x=root;
		while(x!=nil)
		{
			int tem=x->c[0]->sz+1;
			if(tem==k)break;
			else if(tem>k)x=x->c[0];
			else if(tem<k)
			{
				k-=tem;
				x=x->c[1];
			}
		}
		splay(x,fa);
	}
	void insert(int val)
	{
		if(root==nil)
		{
			root=newnode(val,nil);
			return;
		}
		Node *x=root;
		while(1)
		{
			if(val>x->val)
			{
				if(x->c[1]==nil)break;
				x=x->c[1];
			}
			else
			{
				if(x->c[0]==nil)break;
				x=x->c[0];
			}
		}
		if(val>x->val)
		{
			x->c[1]=newnode(val,x);
			x=x->c[1];
		}
		else
		{
			x->c[0]=newnode(val,x);
			x=x->c[0];
		}
		splay(x,nil);
	}
	void delet(int val)
	{
		Node *x=root;
		while(x!=nil)
		{
			if(x->val<val)
			{
				x=x->c[1];
			}
			else if(x->val>val)
			{
				x=x->c[0];
			}
			else
			{
				break;
			}
		}
		splay(x,nil);
		if(root->c[0]!=nil)
		{
			select(root->c[0]->sz,root);
			root->c[0]->c[1]=root->c[1];
			if(root->c[1]!=nil)root->c[1]->fa=root->c[0];
			root->c[0]->fa=nil;
			Node *lroot=root;
			root=root->c[0];
			delete lroot;
			up(root);
		}
		else
		{
			root->c[1]->fa=nil;
			Node *lroot=root;
			root=root->c[1];
			delete lroot;
		}
	}
	int _query(Node *x,int mid)
	{
		if(x==nil)return 0;
		if(mid<x->val)return _query(x->c[0],mid);
		else return x->c[0]->sz+1+_query(x->c[1],mid);
	}
	int num(int mid)
	{
	//	int x=((root->val)%(root->sz)+(root->sz))%(root->sz)+1;
	//	select(x,nil);
		return _query(root,mid);
	}
};
int n,m;
struct T
{
	SPLAY splay;
}tree[MAXN*4+10];
void build(int id,int l,int r)
{
	tree[id].splay.inti(s[l]);
	for(int i=l+1;i<=r;i++)tree[id].splay.insert(s[i]);
	if(l==r)return;
	else
	{
		int mid=(l+r)>>1;
		build(id<<1,l,mid);
		build(id<<1|1,mid+1,r);
	}
}
int query(int id,int tl,int tr,int l,int r,int mid)
{
	if(l<=tl&&tr<=r)
	{
		return tree[id].splay.num(mid);
	}
	else
	{
		int mid1=(tl+tr)>>1;
		int ret=0;
		if(l<=mid1)ret+=query(id<<1,tl,mid1,l,r,mid);
		if(r>mid1)ret+=query(id<<1|1,mid1+1,tr,l,r,mid);
		return ret;
	}
}
void update(int id,int tl,int tr,int pos,int val)
{
	tree[id].splay.delet(s[pos]);
	tree[id].splay.insert(val);
	if(tl==tr)return;
	else
	{
		int mid=(tl+tr)>>1;
		if(pos<=mid)update(id<<1,tl,mid,pos,val);
		else update(id<<1|1,mid+1,tr,pos,val);
	}
}
int main()
{
	freopen("park.in","r",stdin);
	freopen("park.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&s[i]);
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int K;
		scanf("%d",&K);
		if(K==1)
		{
			int a,b,k;
			scanf("%d%d%d",&a,&b,&k);
			int l=-n,r=n;
			k=b-a+2-k;
			while(l<r)
			{
				int mid=(l+r)>>1;
				if(query(1,1,n,a,b,mid)>=k)r=mid;
				else l=mid+1;
			}
			printf("%d\n",l);
		}
		else if(K==2)
		{
			int p,x;
			scanf("%d%d",&p,&x);
			update(1,1,n,p,x);
			s[p]=x;
		}
	}
	return 0;
}


<think>好的,我现在需要帮助用户解决关于蓝桥杯比赛中K倍区间问题。首先,我得回忆一下K倍区间的题目要求。题目概是给定一个数列,要求找出所有子区间的和能被K整除的数量。这个问题通常可以用前缀和和哈希表来优化,避免暴力枚举所有子区间,从而降低时间复杂度。 用户可能已经了解过一些基础的前缀和知识,但需要具体的解题思路和代码实现。我需要先解释清楚前缀和的概念,然后引入同余定理,因为这是解题的关键。同余定理指出,如果两个前缀和对K取余的结果相同,那么这两个前缀和之间的子区间的和一定是K的倍数。这一点需要详细说明,确保用户理解为什么用哈希表来记录余数的出现次数。 接下来,我需要考虑如何构建前缀和数组,并在计算过程中维护一个哈希表,记录每个余数出现的次数。初始时,余数0出现一次,这样可以处理从数组开头到当前位置的子区间刚好是K倍的情况。例如,前缀和数组的第i项和第j项余数相同,那么i到j之间的子区间和就是K的倍数。 然后,我需要给出具体的代码实现。可能用户需要Python版本的代码,因为引用的资料中有Python的例子。代码部分要简洁,同时加上注释,解释每一步的作用。特别是处理负数余数的情况,需要转换为正数,避免余数为负带来的错误。 另外,用户提供的引用中有动态规划解法的提示,我需要确认这与常规的前缀和+哈希表方法是否一致。动态规划在这里可能指的是通过前缀和的状态转移,但核心还是同余定理的应用。需要明确说明这一点,避免混淆。 还需要考虑测试样例,比如用户可能想验证代码的正确性,可以建议他们用示例输入,如题目中的样例,确保输出正确。此外,时间复杂度的分析也很重要,说明该方法为什么高效,通常是O(N)的时间复杂度,适用于数据量的情况。 用户可能的问题还包括如何处理边界情况,比如数组长度为0或者K为0的情况,但题目中K通常是一个正整数。需要指出题目可能的约束条件,确保代码的健壮性。 最后,生成的相关问题需要覆盖性能分析、优化方法和应用场景,帮助用户进一步深入理解。例如,性能分析可以讨论不同方法的时间复杂度对比,优化方法可能包括空间优化或者处理数时的技巧,应用场景可以提到类似的区间统计问题。 总结下来,回答的结构应该是先介绍问题,再详细讲解思路,包括前缀和和同余定理,然后给出代码并解释关键点,最后附上相关问题引导深入学习。确保引用正确的资料,如引用[1]提到的动态规划解法,实际上就是这里使用的方法,需要正确关联。</think>### K倍区间 解题思路 **题目要求**:给定长度为$n$的数列,找出所有连续子区间满足区间和为$K$的倍数的个数。 #### 核心思路 1. **前缀和预处理**:计算前缀和数组$preSum$,其中$preSum[i] = (a_1 + a_2 + ... + a_i) \mod K$ 2. **同余定理应用**:若两个前缀和满足$preSum[i] \equiv preSum[j] \ (\mod K)$,则区间$[i+1,j]$的和是$K$的倍数[^1] 3. **哈希表统计**:使用哈希表记录每个余数出现的次数,通过组合数公式$C(n,2)$计算相同余数的组合可能 #### 关键步骤 - **负数处理**:将负数余数转换为正数,例如$(-7)\%5=3$ - **初始化哈希表**:需预先存入余数0,处理从第一个元素开始就满足条件的情况 ### 代码实现(Python) ```python n, k = map(int, input().split()) count = {0: 1} # 初始化余数0出现一次 pre_sum = 0 res = 0 for _ in range(n): num = int(input()) pre_sum = (pre_sum + num) % k # 计算当前前缀和余数 if pre_sum < 0: # 处理负数余数 pre_sum += k # 累加已有同余组合数 res += count.get(pre_sum, 0) # 更新哈希表 count[pre_sum] = count.get(pre_sum, 0) + 1 print(res) ``` #### 代码解析 1. **输入处理**:读取数列长度$n$和模数$k$ 2. **哈希表初始化**:`count`字典存储余数出现次数,初始状态余数0出现1次 3. **遍历计算**: - 动态维护前缀和余数 - 负数余数转换为正数 - 通过哈希表直接获取能形成K倍区间的组合数 - 更新当前余数的出现次数 #### 复杂度分析 - 时间复杂度:$O(n)$,仅需一次遍历 - 空间复杂度:$O(k)$,哈希表最多存储$k$种余数 ### 示例验证 输入样例: ``` 5 2 1 2 3 4 5 ``` 计算过程: - 前缀和余数序列:1, (1+2)%2=1, (3+3)%2=0, (6+4)%2=0, (10+5)%2=1 - 余数统计:0出现3次,1出现3次 - 结果:$C(3,2)+C(3,2)=3+3=6$
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值