1. 程式人生 > >阿里巴巴2018屆演算法崗招聘題

阿里巴巴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;
}