G. Old Floppy Drive（st表+二分套二分，1900）

ACfunhsl

于 2021-05-14 23:28:08 发布

阅读量178

点赞数

分类专栏：题解

本文链接：https://blog.csdn.net/Joker_He/article/details/116809904

版权

题解专栏收录该内容

279 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何解决一个关于环形磁盘的问题，其中指针按照特定规则行走并累计区域值。文章提出了利用二分查找策略确定满足条件的最少步数，并详细阐述了单调性的证明和前缀和数组结合ST表的处理方法。通过预处理和区间最值查询，实现了O(mlognlogn)的时间复杂度解决方案。

摘要由CSDN通过智能技术生成

题意： 给你一个环形的磁盘，对于磁盘的n个区域，每个区域有一个整数 $a_i$ ，你每次会将指针放到1号位置，它从1走到n，因为是环形，n之后又是1。
他对你进行m次询问，每次询问有一个数x，每当你的指针指向一个区域，会对贡献加上当前区域的值（可重复），当贡献至少为时，指针停下，让你输出指针每次走的步数。

思路： 显然，我们可以对走过的圈数进行二分，因为这显然有单调性。
证明：我们使用 $sum_i$ 记录前i项的和。
如果 $sum_n>0$ ，那么单调性显然成立，因为如果当前满足，显然可以通过试着减少圈数去满足。如果当前不满足，就必须通过增加圈数来满足。
如果 $sum_n<0$ ，那么单调性也成立，因为如果当前满足，减少圈数必满足。如果当前不满足，就要通过减少圈数满足。

在我们二分了圈数之后，就需要考虑剩下的部分怎么处理。
假设我们当前走了 $a$ 圈，那么现在就有 $a*sum_n$ 的贡献，还需要至少 $x-a*sum_n$ 的贡献，那么就是在前缀和数组中找到第一个 $i$ ，使得 $sum_i\ge x-a*sum_n$ ，这时候就用 $s t$ 表维护一下前缀和数组的区间最值，然后二分找第一个合法的。

对 $log_2x$ 进行预处理，st表的查询复杂度降至 $O (1)$ ，代码总的复杂度为 $O (m l o g n l o g n)$
代码：

#define int __int128
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read()
{
	int x = 0,f = 1;
	char ch = gc();
	while(ch < '0' || ch > '9')
	{
		if(ch == '-')
			f = -1;
		ch = gc();
	}
	while(ch >= '0' && ch <= '9')
	{
		x = x * 10 + ch-'0';
		ch = gc();
	}
	return x * f;
}
int st[N][20],a[N],sum[N],lg[N];
int query(int k)
{
	int t = lg[k];
	return max(st[1][t],st[k-(1<<t)+1][t]);
}
void print(int x)
{
	if (!x) return ;
	if (x < 0) putchar('-'),x = -x;
	print(x / 10);
	putchar(x % 10 + '0');
}
signed main()
{
	int t=read();
	for(int i=2; i<N; i++)
		lg[i] = lg[i>>1] + 1;
	while(t--)
	{
		int n=read(),m=read();
		for(int i=1; i<=n; i++) a[i]=read(),sum[i]=sum[i-1]+a[i],st[i][0]=sum[i];
		for(int j=1; (1<<j)<=n; j++)
			for(int i=1; i+(1<<j-1)<=n; i++)
				st[i][j] = max(st[i][j-1],st[i+(1<<j-1)][j-1]);
		while(m--)
		{
			int x=read();
			int l = 0,r = 1e9,ans = 1e18;
			while(l<=r)
			{
				int mid = l+r>>1;
				int y = x - mid * sum[n];
				int ll = 1,rr = n,fg = 0;
				while(ll<=rr)
				{
					int mm = ll + rr >> 1;
					if(query(mm) >= y)
					{
						rr = mm - 1;
						ans = min(ans,mid*n+mm-1);
						fg = 1;
					}
					else
						ll = mm + 1;
				}
				if(fg)
					r = mid - 1;
				else
				{
					if(sum[n]<=0)
						r = mid - 1;
					else l = mid + 1;
				}

			}
			if(ans == 1e18)
				print(-1);
			else
			{
				if(!ans) printf("0");
				else print(ans);
			}
			printf(" ");
		}
		puts("");
	}



	return 0;
}