1. 程式人生 > 實用技巧 >P3612 [USACO17JAN]Secret Cow Code S

P3612 [USACO17JAN]Secret Cow Code S

題意:給定一個序列s,用它能夠生成一個無限長的序列
第一個序列:\(s1 = s\)
第二個序列:\(s2 = s1 + s1'\)
第三個序列:\(s3 = s2 + s2'\)
...
其中\(si'\)表示將\(si\)中的最後一個元素放到最前面形成的新序列。

思路:把一個序列看成前半部分和後半部分,看x在前半部分還是後半部分
在前半部分 => 子問題:前半部分的x位置
在後半部分 => 子問題:前半部分的idx位置,後半部分的x - mid位置 <=> 前半部分的(x - mid == 1 ? mid : x - mid - 1)位置,因為後半部分是由前半部分旋轉得到的。

#include<iostream>
using namespace std;

#define LL long long

string s;
LL x;

char dfs(LL n, LL x){
    if(n == s.size()) return s[x - 1];
    LL mid = n >> 1;
    if(x > mid) return dfs(mid, (x - mid == 1 ? mid : x - mid - 1));
    else return dfs(mid, x);
}

int main(){
    cin >> s >> x;
    
    LL n = s.size();
    while(n < x) n <<= 1;
    
    // n >= x
    
    cout << dfs(n, x);
}