I. Ice Cream Shop

最新推荐文章于 2022-04-28 18:59:32 发布

开学了8

最新推荐文章于 2022-04-28 18:59:32 发布

阅读量197

点赞数

分类专栏： cf之旅文章标签：算法

本文链接：https://blog.csdn.net/luker6/article/details/124402209

版权

cf之旅专栏收录该内容

63 篇文章 2 订阅

订阅专栏

对于每个房子i，二分找打左右两边最近的冰淇淋店，求出距离dis，那么在 $(i - d i s, i + d i s - 1) 这些位置上都可以获得当前房子 i 的人数$
对坐标进行差分，再前缀和，最后求最大值。
在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;

#define ll long long

int n, m;
const int N = 2e7 + 1000, M = 2e5 + 10;
int b[M],num[N];
ll neg[N];
ll ans;

int main(){
	cin >> n >> m;
	for(int i = 1; i <= n; i++) {
		int val;
		cin >> val;
		num[(i-1)*100] = val;
	}
	for(int i = 1; i <= m; i++) cin >> b[i];
	sort(b+1,b+1+m);
	for(int i = 1; i <= n; i++) {
		int pos_r = lower_bound(b+1,b+1+m,(i-1)*100)-b;
		if(pos_r <= m) { // 右 b[pos_r] 有店
			if(pos_r>1) { // 左边b[pos_l] 有店
				int min_dis = min((i-1)*100-b[pos_r-1], b[pos_r]-(i-1)*100);
				neg[(i-1)*100-min_dis+1] += num[(i-1)*100];
				if((i-1)*100 + min_dis+1 < N - 100) 
					neg[(i-1)*100+min_dis+1] -= num[(i-1)*100];
				else 
					neg[N-100] -= num[(i-1)*100];
			}
			else { // l no 
				int min_dis = min(b[pos_r]-(i-1)*100, (i-1)*100);
				neg[(i-1)*100-min_dis+1] += num[(i-1)*100];
				if(b[pos_r] < N-100) neg[b[pos_r]+1] -= num[(i-1)*100];
				else neg[N-100] -= num[(i-1)*100];
			}
		}else { // 右边没有店, 左边必有店
			int min_dis = (i-1)*100 - b[pos_r-1];
			neg[b[pos_r-1]+1] += num[(i-1)*100];
			if((i-1)*100 + min_dis+1 < N - 100)  // 右边界都要判断一下
				neg[(i-1)*100 + min_dis] -= num[(i-1)*100];
			else 
				neg[N-100] -= num[(i-1)*100];
		}
	}
	for(int i = 0; i <= n*100; i++) {
		if(i==0) {
			ans = max(ans,neg[i]);
			continue;
		}
		neg[i] = neg[i-1] + neg[i];
		ans = max(ans, neg[i]);
	}
	cout << ans;
	return 0;
}