【洛谷P4168】蒲公英【分块】

题目大意:

题目链接:https://www.luogu.org/problemnew/show/P4168
给出一个长度为 n n n的序列, m m m次询问,求区间 [ l , r ] [l,r] [l,r]中的众数。强制在线。


思路:

以下内容参考 l y d lyd lyd《算法竞赛进阶指南》。

先离散化不解释。
分块算法一般都是以“暴力+区间预处理+暴力”的方法做的。所以,我们把 n n n分成 T T T块,每块长度 ⌊ T n ⌋ ⌊\frac{T}{n}⌋ nT
然后预处理出对于任意两个区间内的数字个数。也就是说,对于任意的 1 ≤ l ≤ r ≤ T 1\leq l\leq r\leq T 1lrT,我们用 c n t [ L ] [ R ] [ i ] cnt[L][R][i] cnt[L][R][i]表示块 L ∼ R L\sim R LR内数字 i i i的个数。
如何预处理出这一部分?

  • 方法1
    先暴力处理出所有的 c n t [ i ] [ i ] [ j ] cnt[i][i][j] cnt[i][i][j],然后用区间 d p dp dp的思想,求出所有的 c n t [ L ] [ R ] [ j ] cnt[L][R][j] cnt[L][R][j],时间复杂度 O ( T 2 n ) O(T^2n) O(T2n)
  • 方法2
    预处理出 s u m [ R ] [ i ] sum[R][i] sum[R][i]表示区间 1 ∼ R 1\sim R 1R之间 i i i的个数。然后用前缀和思想, c n t [ L ] [ R ] [ i ] = s u m [ R ] [ i ] − s u m [ L ] [ i ] cnt[L][R][i]=sum[R][i]-sum[L][i] cnt[L][R][i]=sum[R][i]sum[L][i]。时间复杂度 O ( T 2 ) O(T^2) O(T2)

同时在预处理 c n t cnt cnt的时候顺便处理出 M a x [ L ] [ R ] Max[L][R] Max[L][R],表示区间众数,然后记录众数的个数。
接下来就是如何处理答案了。
对于任意一组询问,设询问区间 [ l , r ] [l,r] [l,r],那么先记录 q , p q,p q,p分别表示 l , r l,r l,r所在的块。如果 p − q ≤ 1 p-q\leq 1 pq1,那么这两个块中间就没有多余的整块,直接朴素求众数。
否则在 c n t [ q + 1 ] [ p − 1 ] cnt[q+1][p-1] cnt[q+1][p1]的整块区间中,加上 [ l , R [ q ] ] , [ L [ p ] , r ] [l,R[q]],[L[p],r] [l,R[q]],[L[p],r]之间的数字,然后求出答案。最后再减去加上的数就可以了。


算法时间复杂度为 O ( n T 2 + m T n ) O(nT^2+\frac{mT}{n}) O(nT2+nmT),空间复杂度为 O ( n T 2 ) O(nT^2) O(nT2)。为了平均时间,不妨设 n T 2 = m T n nT^2=\frac{mT}{n} nT2=nmT,解得 T ≈ n 3 T≈\sqrt[3]{n} T3n 。此时时间复杂度约为 O ( n 5 3 ) O(n^{\frac{5}{3}}) O(n35)


代码:

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;

const int N=50010;
int a[N],b[N],cnt[40][40][N],L[40],R[40],pos[N],sum[N],Max[40][40][2],be[N];
int n,m,T,x,y,len,tot,last;

int ask(int l,int r)
{
	int ans=0,maxn=0;
	int q=pos[l],p=pos[r];
	if (p-q<=1)
	{
		memset(sum,0,sizeof(sum));
		for (int i=l;i<=r;i++)
		{
			sum[a[i]]++;
			if (sum[a[i]]>maxn||(sum[a[i]]==maxn&&a[i]<ans))
				maxn=sum[a[i]],ans=a[i];
		}
		return ans;
	}
	maxn=Max[q+1][p-1][0];
	ans=Max[q+1][p-1][1];
	for (int i=l;i<=R[q];i++)
	{
		cnt[q+1][p-1][a[i]]++;
		if (cnt[q+1][p-1][a[i]]>maxn||(cnt[q+1][p-1][a[i]]==maxn&&a[i]<ans))
			maxn=cnt[q+1][p-1][a[i]],ans=a[i];
	}
	for (int i=L[p];i<=r;i++)
	{
		cnt[q+1][p-1][a[i]]++;
		if (cnt[q+1][p-1][a[i]]>maxn||(cnt[q+1][p-1][a[i]]==maxn&&a[i]<ans))
			maxn=cnt[q+1][p-1][a[i]],ans=a[i];
	}
	for (int i=l;i<=R[q];i++) cnt[q+1][p-1][a[i]]--;
	for (int i=L[p];i<=r;i++) cnt[q+1][p-1][a[i]]--;
	return ans;
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		b[i]=a[i];
	}
	sort(b+1,b+1+n);
	tot=unique(b+1,b+1+n)-b-1;
	for (int i=1;i<=n;i++)
	{
		x=lower_bound(b+1,b+1+tot,a[i])-b;
		be[x]=a[i];
		a[i]=x;
	}
	T=(int)pow((double)n,1.0/3.0);
	len=n/T;
	if (T*len<n) T++;
	for (int i=1;i<=T;i++)
	{
		L[i]=R[i-1]+1;
		R[i]=min(i*len,n);
		for (int j=L[i];j<=R[i];j++)
		{
			cnt[i][i][a[j]]++;
			if (cnt[i][i][a[j]]>Max[i][i][0]||(cnt[i][i][a[j]]==Max[i][i][0]&&a[j]<Max[i][i][1]))
			{
				Max[i][i][0]=cnt[i][i][a[j]];
				Max[i][i][1]=a[j];
			}
			pos[j]=i;
		}
	}
	for (int k=1;k<T;k++)
		for (int i=1;i<=T-k;i++)
		{
			int j=i+k;
			int mid=(i+j)/2;
			for (int l=1;l<=tot;l++)
			{
				cnt[i][j][l]=cnt[i][mid][l]+cnt[mid+1][j][l];
				if (cnt[i][j][l]>Max[i][j][0])
				{
					Max[i][j][0]=cnt[i][j][l];
					Max[i][j][1]=l;
				}
			}
		}
	while (m--)
	{
		scanf("%d%d",&x,&y);
		x=(x+last-1)%n+1;
		y=(y+last-1)%n+1;
		if (x>y) swap(x,y);
		last=be[ask(x,y)];
		printf("%d\n",last);
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值