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 的翻译

 

 

刚开始无脑常规写法，只得了40分，因为数据范围在10^18内，超过了字符串的长度，内存爆了。

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
int main()
{
    string s,s1;
    ll n,len,num;
    cin>>s>>n;
    num=log(n/s.size())/log(2)+1;
    for(int i=1; i<=num; i++){
       s1=s.substr(0,s.size()-1);
       s1=s[s.size()-1]+s1;
       s+=s1;
    }
    cout<<s[n-1]<<endl;
    return 0;
}

//无脑写法

正确思路：分治、递归

设原字符串的长度为s

若n<=s,可以直接返回答案

若n>s，则去求n所对应的规模更小的位置n'(重点在此)

1.求第一个>=n的长度len

2. 取len的中点，若n>mid，如下：

src="ABCD" ，len=8, n=6,mid=4， 且n-mid!=1

ABCDDABC

123456789

求n'. 因为6>4,即n>mid, 事实上n'=1=n-mid-1,减1是去掉被旋转到串头的字符。

src="ABCD" ，len=8, n=7,mid=4， 且n-mid==1,说明该字符正好是字符串的最后一个字符。

3. 若n<=mid, 那么递归将前面1,2再做一遍。

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
string s;
char fun(ll x,ll l, ll n){//x原字符串长度
    ll mid;
    if(n<=x){
       return s[n-1];
    }
    mid=l>>1;
    if(n>mid){//右边求对应位置
        if(n-mid==1)//第一个对应原来最后一个
          fun(x,mid,mid);
        else fun(x,mid,n-mid-1);
    }
    else {//左边求对应位置
        ll len=x;
        while(len<n) len*=2;
        fun(x,len,n);
    }
}
int main()
{
    ll n,len,num;
    cin>>s>>n;
    len=s.size();
    while(len<n){
       len*=2;
    }
    cout<<fun(s.size(),len,n)<<endl;

    return 0;
}