阿里巴巴2018届算法岗招聘题

考试的时候通过率60%，找了半天bug，没发现错误，以为int大小不够，到交卷了才反应过来，程序有时间限制，改进算法如下，时间比60%好，事后诸葛

；

#include<iostream>  

#include<cmath>

using namespace std;

long int wz(long int n)

{

    long int i=1;
while ( !(n<pow(10,i) ) )
{
i++;
}

    return i;

}

long int Get(long int n)

{

    long int x;

long int i,k=1,sum=0,bz=1,sum0=0;

while(bz)
{
sum0+=wz(k);

        sum+=sum0;
if (sum>=n)
{
 bz=0;
 break;
}
   k++;
}

  
int temp;
if(sum==n)
{
x=k%10;
}
else
{

for(i=k;i>=1;i--)
{
   sum-=wz(i);
if (sum<n)
{
   break;
}
}

        sum+=wz(i);

temp=sum-n;
   for(k=1;k<=temp;k++)
{
       i/=10;
}

        i%=10;
x=i;
}

    

        return x;

}

int main()  

{  

    long int n;

    cin>>n;

    long int r = Get(n);

        

    cout<<r;  

    return 0;

}