CodeForces 820D Mister B and PR Shifts（思维题）

最新推荐文章于 2020-12-02 14:38:51 发布

ccDLlyy

最新推荐文章于 2020-12-02 14:38:51 发布

阅读量643

点赞数 1

分类专栏：思想+思维 codeforces

本文链接：https://blog.csdn.net/ccDLlyy/article/details/78471099

版权

codeforces 同时被 2 个专栏收录

20 篇文章 0 订阅

订阅专栏

思想+思维

15 篇文章 0 订阅

订阅专栏

博客讲述了如何高效解决CodeForces上的一个问题，即寻找一个排列在循环右移过程中的最小值及其出现次数。作者最初尝试了O(N^2)的解决方案，但发现超时。通过研究其他代码，找到了一个O(N)的思路，利用cnt数组记录每个数移到其实际下标所需移动次数，并用add和sub变量动态维护移位后sum值的变化。关键在于理解每个数的位置对于移位后绝对差值的影响，特别是位置num[i]是变化的转折点。

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

题意：给定一个长度为n的排列，每次向右循环移位，求的最小值，并求出是在第几次移位得到的。

思路：开始O（N^2）模拟了一下，果然超时。参考了一下网上代码，知道了一个O（N）的思路。用cnt数组记录当前num[i]所在位置到下标为num[i]的位置需要向右循环移位几次，两个变量add、sub表示下一次移位会使sum值+1/-1的个数，每次更新均在上一排列的基础上进行。

先注意题意， 1 <= num[i] <= n。

// CodeForces 820D Mister B and PR Shifts 运行/限制：264ms/2000ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;
#define INF 0x3f3f3f3f
#define LL long long
int num[1000005],cnt[1000005];
int main(){
	int n;
	int add, sub, index;
	LL sum, re;
	while (scanf("%d", &n) != EOF) {
		sum = 0; add = 0; sub = 0;
		memset(cnt, 0, sizeof(cnt));
		for (int i = 1; i <= n; i++) {
			scanf("%d", &num[i]);
			sum += fabs(num[i] - i);
			cnt[(num[i] - i + n) % n]++;//cnt下标为：当前num[i]所在位置 到 num[i]位置 需要移位几次
			if (num[i] > i) sub++;//下一次移位会使值-1的个数
			else add++;//下一次移位会使值+1的个数
		}
		re = sum; index = 0;
		for (int i = 1; i < n; i++) {//每次在前一次移位的基础上处理，i为第几次移动
			sum += add - sub - 1;//当前最后一位肯定满足数值小于等于n（数据范围最大是n），所以之前计算在add里了；但因为要移动到首位，所以不一定是加、减
			sum += (num[n - i + 1] - 1) - (n - num[n - i + 1]);//计算当前最后一位移动到首位sum值的变化；
			                                                   //每次并没有真正进行移位，而是每次计算最后一位应该是未移位序列的哪个位置了

			//求完sum，相当于该次移动了

			//根据此次移动后的结果，更新下一步add、sub会出现的值
			//-1,+1是因为这一次移动后到第一个位置的数，由于单调性反转，使add比正常情况-1、sub+1。
			//num[i] = 1时，|num[i] - loc|全程单调递增，从n位置移动到1位置时单调性的变化是个特例，可以分析一下num[n] = 1也满足下面式子
			add = add + cnt[i] - 1;
			sub = sub - cnt[i] + 1;
			if (sum < re) {
				re = sum;
				index = i;
			}
		}
		printf("%lld %d\n", re, index);
	}
    return 0;
}