#include<iostream>
using namespace std;
/*
蓝桥杯: 杨辉三角形
给出一个数字n,说出它第一次出现在杨辉三角形的第几个数
如:
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
将其按行读取,排序为: 1 1 1 1 2 1 1 3 3 1 1 4 6 4 1
输入6,那么输出为: 13
对于20%的测试数据: 1<= N <=10
对于100%的测试数据: 1<= N <= 100000000
[分析]
1 1
1 1 1
1 2 1 1 2
1 3 3 1 1 3
1 4 6 4 1 1 4 6
1 5 10 10 5 1 1 5 10
1 6 15 20 15 6 1 1 6 15 20
1 7 21 35 35 21 7 1 1 7 21 35
1 8 28 56 70
图1 图2
我们的策略是按列找,按行找会很难找,图二只取杨辉三角形的对称部分,下面对图二讨论.
第i列第一个数为(i从0开始): (2i * 2i-1 * ... * i+1) / i!
第i列第 k个数为(i从0开始,k从1开始): (2(i+k-1) *2(i+k-1)-1 * ... * 2(i+k-1)-i+1 ) / i!
即相当于概率论中的 "C" 上标为 i ,下标为 2i+k-1 ,见下面的截图,打字打不出来
那么我们要找的数最大为100000000 当 i = 17时, 也即 17列时出现的第一个数是大于100000000,
那么我们从16行往下找起, 直到找到等于它的数 . 当然也可能找不到与之相同的数 ,我们找到第一个比他大的数时, 则说明这一列不存在与之相同的数了, 我们换下一列继续寻找
这样一直找下去, 我们会找到的第一个与之相等的数, 一定是行数最少的数, 因为如果前面一列也出现这个数, 那么设第一次找到该数为第i行,在前面某一列再次找到这个数出现的
行为第j行,j一定大于i.
行数越小, 说明出现的越早.
找到行数和列数,出现的位置也能相应推算出来了.
不知道为什么只拿了80分,理论上应该是不会超时的,但是有两个测试数据超时了
*/
int C(int up,int down); // 概率论中 "C" 运算 up:上标 down:下标
long long int AddSum(int n); //计算从1 + 2 + ... + n ,累加的最大值是1 + 2 + ... + 100000001,必须用long long int
int num;
int main()
{
int i,j;
int hang,lie;
cin>>num;
bool flag = true;
for(i=16;i>=0&&flag;i--){
int firstNum = C(i,2*i);
int count = 1;
while(firstNum<=num){
if(firstNum==num){
lie = i;
hang = count; //这里的行不是实际意义上的行 ,我们的列是从0开始,行是从1开始
flag = !flag;
break;
}
firstNum = C(i,2*i+count);
count ++;
}
}
hang = lie*2 + hang; //实际在杨辉三角形上的行
cout<<AddSum(hang-1) + lie + 1 <<endl;
return 0;
}
int C(int up,int down){
int returnValue = 1;
int i,j;
for(i=down,j=1;j<=up;i--,j++){
returnValue = returnValue * i / j;
if(returnValue>num){ //不用算了因为再怎么算也要大于num,算下去可能会数据越界直接gg
break;
}
}
return returnValue;
}
long long int AddSum(int n)
{
int i=1;
long long int sum = 0;
for(i=1;i<=n;i++){
sum += i;
}
return sum;
}
学习笔记