2021年度训练联盟热身训练赛第四场 I.Slot Machines

题意:
给定一个长度为 n n n的序列 a a a,让你找出一对 k k k p p p使得对于 i > k i>k i>k
都有 a [ i ] = a [ i + p ] a[i]=a[i+p] a[i]=a[i+p],要求使得 k + p k+p k+p最小,当多个 k + p k+p k+p相同时, p p p尽可能小。
数据范围: 1 ≤ n ≤ 1 0 6 1\leq n\leq 10^6 1n106

题解:
先将序列翻转过来得到 a r ar ar,那么循环节元素必然是从 a r [ 1 ] ar[1] ar[1] a r [ p ] ar[p] ar[p]
所以我们需要知道如何快速判断,考虑到循环节的特性,求解 n e x t next next数组。
考虑枚举每个元素作为其所在循环节的最后一个元素,
完整的循环节长度 p = i − n e x t [ i ] p=i-next[i] p=inext[i],非循环节部分长度为 k = n − i k=n-i k=ni
这部分完整循环节为: a r [ i − n e x t [ i ] + 1 , i ] ar[i-next[i]+1,i] ar[inext[i]+1,i]
理解的方式:序列为未翻转的状态,那么可以知道 a [ k + 1 , k + i − n e x t [ i ] ] a[k+1,k+i-next[i]] a[k+1,k+inext[i]]就是翻转的 a r [ i − n e x t [ i ] + 1 , i ] ar[i-next[i]+1,i] ar[inext[i]+1,i]

因此这样我们只需要对每次枚举得到的 p , k p,k p,k对最终答案更新即可。由于 k k k枚举时递减,所以当 p + k p+k p+k相同时,越往后枚举得到的 p p p必然越大,因此只有 p + k p+k p+k小于当前答案才能更新。

代码:

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

const int N = 1e6 + 10;
int a[N], n;
int ne[N];
void get_next() {
	for(int i = 2, j = 0; i <= n; ++i) {
		while(j && a[i] != a[j + 1]) j = ne[j];
		if(a[i] == a[j + 1]) ++j;
		ne[i] = j;
	}
}

int main()
{
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	reverse(a + 1, a + n + 1);
	get_next();
	
	int p = 1, k = n - 1;
	for(int i = 2; i <= n; ++i) {
		int tp = i - ne[i];
		int tk = n - i;
		if(tp + tk < p + k) p = tp, k = tk;
	}
	printf("%d %d\n", k, p);
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值