思路1:
有杨辉三角排列规律得,每个数字等于它上方两个数字之和;即dp[i][j]=dp[i-1][j-1]+dp[i-1][j],那么按照这种递推方法代码如下:
public class YangHui {
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
int n=scanner.nextInt();
int m=scanner.nextInt();
long[][] dp=new long[n][n];
int i=0;
int j=0;
for(int k=0;k<n;k++)
{
dp[k][0]=1;
dp[k][k]=1;
}
for( i=2;i<n;i++)
{
for( j=1;j<m;j++)
{
dp[i][j]=dp[i-1][j-1]+dp[i-1][j];
}
}
System.out.println(dp[n-1][m-1]);
}
}
但是上种方式计算每个数字需要依赖上一行两个数字,时间复杂度大,容易超时。
思路2:卢卡斯算法
1 第一行第一个数字 C(0,0)=1
1 1 第二行第一个数字C(1,0)=1 第二行第二个数字 C(1,1)=1
1 2 1 第三行第一个数字C(2,0)=1 ,第3行第二个数字C(2,1)=2 第三行第三个数字C(2,2)=1
1 3 3 1 .....
1 4 6 4 1 ....
.......
可以得出在杨辉三角中,第n行第m列数字可以用排列组合表示:C(n-1,m-1)。
但是n,m较大的话,阶乘会很大,所以采用卢卡斯算法处理这种大组合,公式为:C(n,m)%p(p一定是素数)
代码如下:
public class Yanghui1 {
public static final int p=1000000007; //p是素数
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
int n=scanner.nextInt();
int m=scanner.nextInt();
int a=n-1;
int b=m-1;
long ret=Lucas(a,b);
System.out.println(ret);
}
public static long Lucas(long n,long m)
{
if(m==0)
return 1;
return (Lucas(n/p,m/p)*C(n%p,m%p))%p;
}
public static long C(long n,long m)
{
if(n<m)
{
return 0;
}
long ret=1;
for(int i=1;i<=m;i++)
{
ret=ret*(((n-m+i)%p)*Pow(i,p-2,p)%p)%p;
}
return ret;
}
public static long Pow(long a,long b,long mod)
{
long ret=1;
while(b!=0)
{
if((b&1)!=0)
{
b--;
ret=(ret*a)%mod;
}
else
{
b/=2;
a=(a*a)%mod;
}
}
return ret;
}
}