HDU 4333 Revolving Digits (exkmp)

题意
给你一个数 N N 大小为 1 0 100000 10^{100000} ,之后你可以将这个数字的最后一位放到第一位其他位置顺次往后移动,现在问你,在移动的过程中有多少个数字大于 N N 或者小于 N N ,或者等于 N N .
思路
字符串特别大,所有显然要用字符串输入。首先我们将 N N 翻一倍续到后面记为 s h sh ,这样的话,新的字符串就会包含字符串变换的所有情况。之后我们发现每次变换之后我们要和原来的串 N N 来比较,具体怎么比较,就是先比较第一位如果第一位就已经比出结果了那么就结束,不然就依次的去比较下一位,但是显然这样一个一个的比较很慢,所以仔细想一下我们能不能提前预处理出 s h sh 后面的每一位和比较的串 N N 在何时相等,那么我们直接跳过相等的部分直接去比较看他们哪一位不相等就好了,那么其实就是求出 s h sh 的所有后缀和 N N 的前缀匹配的长度,所以求一遍exkmp就好了
代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100000 + 10;
string ch , sh;
int nex[maxn] ,ex[maxn];
void pre_kmp()
{
	memset(nex,0,sizeof(nex));
	int len = ch.size();
	int i = 0 , k = -1;
	nex[0] = - 1;
	while(i < len)
	{
		if(k == -1 || ch[i] == ch[k])
		{
			i++,k++;
			nex[i] = k;
		}
		else k = nex[k];
	}
}
void exkmp()
{
	memset(ex,0,sizeof(ex));
	int lens = sh.size();
	int j = 0 , k = 0;
	ex[0] = lens;
	while(j + 1< lens && sh[j] == sh[j+1]) j++;
	ex[1] = j ;
	k++;
	for(int i = 2 ; i < lens ; i ++)
	{
		int p = ex[k] + k - 1;
		int L = ex[i-k];
		if(i+L < p + 1) ex[i] = L;
		else 
		{
			j = max(0,p-i+1);
			while(i+j < lens && sh[i+j] == sh[j]) j ++;
			ex[i] = j ; k = i;
		}
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	for(int T = 1 ; T <= t ; T++)
	{
		cin>>ch;
		pre_kmp();
		int len = ch.size() , cur = 1;
		if(len % (len - nex[len]) == 0) cur = len / (len - nex[len]); // 这里可能这个串是有循环节的,所以要求一下循环节。
		sh = ch + ch; 
		exkmp();
		int L = 0 , G = 0 ,E = 0;
		for(int i = 0 ; i < len  ; i++)
		{
			if(ex[i] >= len) E++;
			else if(sh[ex[i]]>sh[i+ex[i]]) L++;
            else if(sh[ex[i]]<sh[i+ex[i]]) G++;

		}
		printf("Case %d: %d %d %d\n",T,L/cur,E/cur,G/cur);
	}
}

猜你喜欢

转载自blog.csdn.net/wjmwsgj/article/details/83275951
今日推荐