90分求调T_T

LioHein @ 2024-12-16 18:33:28

#include <iostream>
#include <cmath>
#include <string>
using namespace std;

long long N;

int main()
{
    long long index = 0,res=0;
    string a;
    cin >> a >> N;
    int l = a.length();
    a = " " + a;
    if(N<=l) index = -1;
    else
        while( !( (l * pow(2, index)) <= N && N <= l * pow(2, index + 1) ) )
        index++;
    //如果N不在范围内，则index递增，直到N属于范围内

    //原始代码
    /*for(int i=0;i<=index;i++)
        for(int j =l+1 ;j <= l * pow(2, i+1) ; j++){
            if(j == l*pow(2,i) + 1)a[j] = a[j-1];
            else a[j] = a[j - l * pow(2, i) - 1];
        }*/

    //优化后：
        res=N;
        long long j;
    for(int i=index;i>=0;i--){
        if( (l * pow(2, i)) <= res && res <= l * pow(2, i + 1) )
        for(j = res ;j >= l * pow(2, i) ; ){
            if(j == l*pow(2,i) + 1) {j = j - 1;break;}
            else j = (j - l * p ow(2, i) - 1);
            //cout<<'1'<<' '<<j<<endl;检查
        }
        else continue;
        cout<<i<<' '<<j<<endl;
        res = j;
        if(res<=l)break;
    }
    int ans = res;
    cout<<a[ans];
    return 0;
}

by TODAYS @ 2024-12-16 18:57:50

// pow函数返回值是浮点数，有浮点数误差，建议手写函数
#include <iostream>
#include <cmath>
#include <string>
using namespace std;

long long pow_w(int a, int n)
{
    long long sum = 1;
    for (int i = 1; i <= n; i++) sum *= a;
    return sum;
}

long long N;

int main()
{
    long long index = 0,res=0;
    string a;
    cin >> a >> N;
    int l = a.length();
    a = " " + a;
    if(N<=l) index = -1;
    else
        while( !( (l * pow_w(2, index)) <= N && N <= l * pow_w(2, index + 1) ) )
        index++;
    //如果N不在范围内，则index递增，直到N属于范围内

    //原始代码
    /*for(int i=0;i<=index;i++)
        for(int j =l+1 ;j <= l * pow(2, i+1) ; j++){
            if(j == l*pow(2,i) + 1)a[j] = a[j-1];
            else a[j] = a[j - l * pow(2, i) - 1];
        }*/

    //优化后：
        res=N;
        long long j;
    for(int i=index;i>=0;i--){
        if( (l * pow_w(2, i)) <= res && res <= l * pow_w(2, i + 1) )
        for(j = res ;j >= l * pow_w(2, i) ; ){
            if(j == l*pow_w(2,i) + 1) {j = j - 1;break;}
            else j = (j - l * pow_w(2, i) - 1);
            //cout<<'1'<<' '<<j<<endl;检查
        }
        else continue;
//        cout<<i<<' '<<j<<endl;
        res = j;
        if(res<=l)break;
    }
    int ans = res;
    cout<<a[ans];
    return 0;
}