#702 (Div. 3) G. Old Floppy Drive(思维+二分

G. Old Floppy Drive
题意:
给定一个循环数组和 x x x,从第一位开始累加,直到 s u m > = x sum >= x sum>=x 为止,问需要累加多少次或者永远不能达到
思路:
先求一个数组和 s u m sum sum 和最大前缀和 p r e _ m a x pre\_max pre_max
如果满足 p r e _ m a x < x   & &   s u m < = 0 pre\_max<x \ \&\& \ sum <= 0 pre_max<x && sum<=0,则无解
然后 p r e _ m a x pre\_max pre_max s u m sum sum 满足一个关系, p r e _ m a x > = s u m pre\_max >= sum pre_max>=sum
为了使答案尽量小,显然我们应该尽量让最后一次循环截止到 p r e _ m a x pre\_max pre_max 的及其之前的位置
然后先执行 c n t = ( x − p r e _ m a x + s u m − 1 ) / s u m cnt = (x-pre\_max+sum-1)/sum cnt=(xpre_max+sum1)/sum 次,最后剩下 x − c n t ∗ s u m x-cnt*sum xcntsum
剩下的部分我们可以考虑二分查找
显然直接求得的前缀和是不单调的,即使我们加上 i d id id 后对前缀和进行排序,也无法得到最优解
因为们要查找应该是满足 p r e _ m a x > = x pre\_max>=x pre_max>=x 的前缀和中最 i d id id 最小的一个
参考这个大佬的思路
可以发现如果 p r e [ i ] > = p r e [ i + 1 ] pre[i]>=pre[i+1] pre[i]>=pre[i+1],那么 p r e [ i + 1 ] pre[i+1] pre[i+1] 显然永远不会作为答案的,这是一个无效点,因为有更优的 p r e [ i ] pre[i] pre[i]
因此可以考虑维护一个单调递增的 p r e pre pre 数组,这时就可以二分得到最优解
code:

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define ull unsigned long long
#define ld long double
#define all(x) x.begin(), x.end()
#define eps 1e-6
using namespace std;
const int maxn = 2e5 + 9;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
ll n, m;

void work()
{
	scanf("%lld %lld", &n, &m);
	ll sum = 0;
	vector <ll> pre, id;
	for(int i = 1, x; i <= n; ++i){
		scanf("%d", &x);
		sum += x;
		if(pre.empty() || pre.back() < sum)
			pre.push_back(sum), id.push_back(i);
	}
	while(m--){
		int x;scanf("%d", &x);
		if(pre.back() < x && sum <= 0) printf("-1 ");
		else
		{
			ll cnt = 0;
			if(pre.back() < x) cnt = (x - pre.back() + sum - 1) / sum;
			x -= cnt * sum;
			int pos = lower_bound(pre.begin(), pre.end(), x) - pre.begin();
			printf("%lld ", cnt * n + id[pos] - 1);
		}
	}
	printf("\n");
}

int main()
{
//	ios::sync_with_stdio(0);
	int TT;cin>>TT;while(TT--)
	work();
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值