P3612 [USACO17JAN]Secret Cow Code S【递推】

题目描述

The cows are experimenting with secret codes, and have devised a method for creating an infinite-length string to be used as part of one of their codes.

Given a string ss, let F(s)F(s) be ss followed by ss "rotated" one character to the right (in a right rotation, the last character of ss rotates around and becomes the new first character). Given an initial string ss, the cows build their infinite-length code string by repeatedly applying FF; each step therefore doubles the length of the current string.

Given the initial string and an index NN, please help the cows compute the character at the NNth position within the infinite code string.

奶牛正在试验秘密代码，并设计了一种方法来创建一个无限长的字符串作为其代码的一部分使用。

给定一个字符串，让后面的字符旋转一次（每一次正确的旋转，最后一个字符都会成为新的第一个字符）。也就是说，给定一个初始字符串，之后的每一步都会增加当前字符串的长度。

给定初始字符串和索引，请帮助奶牛计算无限字符串中位置N的字符。

输入格式

The input consists of a single line containing a string followed by NN. The string consists of at most 30 uppercase characters, and N \leq 10^{18}N≤1018.

Note that NN may be too large to fit into a standard 32-bit integer, so you may want to use a 64-bit integer type (e.g., a "long long" in C/C++).

第一行输入一个字符串。该字符串包含最多30个大写字母，并 N \leq 10^{18}N≤1018 。

第二行输入N。请注意，数据可能很大，放进一个标准的32位整数可能不够，所以你可能要使用一个64位的整数类型（例如，在C / C++ 中是 long long）。

输出格式

Please output the NNth character of the infinite code built from the initial string. The first character is N=1N=1.

请输出从初始字符串生成的无限字符串中的位置的字符。第一个字符是 N=1N=1.。

输入输出样例

输入 #1复制

COW 8

输出 #1复制

C

说明/提示

In this example, the initial string COW expands as follows:

COW -> COWWCO -> COWWCOOCOWWC

12345678

感谢@y_z_h 的翻译

解题思路

这道题的代码并不长，不过在做之前得好好分析。

这里我们采用递推的思想：

首先，假定len为给定字符串的长度，我们得确定最大要增加到多长才能满足我们取到n位置的字符。对于n，我们需要不断取tmp = tmp*2（tmp初始值等于len）直到len>=n。

然后，我们知道长度为tmp的字符串是由长度为tmp/2的字符串通过一定规则转化而来的。根据这个规则，我们知道当前字符串（下标从0开始）str[tmp/2-1] = str[tmp/2]，且str[0]~str[tmp/2-2] == str[tmp/2+1]~str[tmp-1]。而且n一定是大于tmp/2的，即在tmp的后半段。

根据上面分析，我们可以推出n位置上的字符在前半段字符上的对应位置是：n = n-1-tmp/2，但是如果n-1-tmp/2 == 0 的话，当前n位置上的字符串应该在前半段的字符串的末尾，即n = tmp/2。

然后通过上面方法一直反复执行，知道将当前n缩小到len以内。

 

代码

#include<cstdio>
#include<iostream>
#include<string>
using namespace std;
#define ll long long

ll n, len, tmp;

int main()
{
	string s;
	cin >> s >> n;
	len = s.length();
	while (len<n)
	{
		tmp = len;
		while (n > tmp)
			tmp *= 2;
		tmp /= 2;
		n -= (tmp + 1);
		if (n == 0)
			n = tmp;
	}
	cout << s[n-1] << endl;
	return 0;
}