- 对输入进行简单的判断;
- 根据杨辉三角在二项式定理中的应用计算每层的值也可用数组存储但前者在内存方面有很大优势,在得出元素速度方面55开;(这里无需太过纠结);二项式定理
- 核心思路为两种剪枝情况和由于杨辉三角的对称而只需折半查找(查找)
(3.1)在某一行的计算中如果有一个元素大于了需要查找的number,则这行可以结束了(做到这个剪枝则可在不考虑内存的情况下拿到80分)ps:本人采用的二项式定理由公式得来所以内存用的基本都不到4mb;
(3.2)最关键的剪枝:如果一层中的第三个元素就大于了需要查找的number则说明number需要在第number层才能找到了,此时利用等差数列前n项和得出位置(可能有点抽象,看图)
思路分析完毕代码如下:
#include<stdio.h>
#include<time.h>
long long int number,index=6;
int func(int n){
long long int num=1;
int temp=1;
for(int i=1;i<=n/2;i++){
num*=n-i+1;
num/=i;
temp++;
if(num==number){
return temp;
}
if(num>number){/*剪枝如果大于就没必要继续算了*/
return 0;
}
}
return 0;
}
long long int select(int n){
if(n==1){
return 1;
}else if(n==2){
return 5;
}
int dep=4;
long long int temp;
while(1){
if((dep-1)*(dep-2)/2>number){/*最关键的剪枝*/
return (number+1)*number/2+2;
}
temp=func(dep-1);
if(temp==0){/*没找到*/
index+=dep;
}else{/*找到了*/
index+=temp;
return index;
}
dep++;
}
}
int main(){
scanf("%d",&number);
long long int t1,t2;
printf("%lld",select(number));
}
ps:如有bug 或优化方案请各路大佬不吝赐教,如有疑问可评论看到必答;