E - Chinese Restaurant abc268

原创已于 2023-02-27 12:47:48 修改 · 308 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #算法 #图论

于 2022-09-22 21:35:28 首次发布

题目同时被 2 个专栏收录

6 篇文章

订阅专栏

Atc

3 篇文章

订阅专栏

本文介绍了一种解决n个人围坐一桌，每个人有特定菜品喜好的问题，通过分析每个人的沮丧值变化，利用算法找出调整座位使得所有人满意度最大化的方法。核心是将函数分为线性部分并独立处理系数，最终给出最小沮丧值的计算过程。

link
题意：n个人围成一桌，编号0 到 n-1, 每个人有自己喜欢的菜品，每个人收到的沮丧值是喜欢的菜品距自

思路：
考虑单独一个人的沮丧值关系，如果顺指针移动桌子，那么第（p[i] - i + n） % n 次此人沮丧值为零。
即为函数的零点。函数长这样
零点在左
或者在中间
或在右边
最多分成三段（一次函数），我们将系数独立起来，然后累加。

code:

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 4 * 1e5 + 10, mod = 998244353;
int n, m;
int sk[maxn], sb[maxn], p[maxn], cst[maxn];
void add(int l, int r, int k, int b) {
	if (l > r) return;
	sk[l] += k;
	sb[l] += b;
	sk[r + 1] -= k;
	sb[r + 1] -= b;
}
void add(int v) {
	if (v < n / 2) {
		add(0, v - 1, -1, v);
		add(v, v + n / 2, 1, -v);
		add(v + n / 2 + 1, n - 1, -1, n + v); // n + v 尤其精妙，分奇偶进行讨论，合并后就是n + v; 
	} else {
		add(v - n / 2, v, -1, v);
		add(v + 1, n - 1, 1, -v);
		add(0, v - n / 2 - 1, 1, n - v);
	}
}
signed main() {
	//freopen("w.ans", "w", stdout);
	ios::sync_with_stdio(false);
	cin.tie(0);
	//int tt; cin >> tt; while(tt--) solve(); 
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> p[i];
		cst[i] = (p[i] - i + n) % n;
		add(cst[i]);
	}
	for (int i = 1; i < n; i++) {
		sb[i] += sb[i - 1];
		sk[i] += sk[i - 1];
	}
	int ans = 1e15, sum = 0;
	for (int i = 0; i < n; i++) {
		ans = min(ans, sk[i] * i + sb[i]);
	}
	cout << ans << endl;
	return 0;
/**
 *    Alogrithm:
 *    201
 *    author:  whc
 *    created: 2022年9月12日10:58:05 
**/
}