码迷,mamicode.com
首页 > 其他好文 > 详细

Educational Codeforces Round 80 C. Two Arrays(组合数快速取模)

时间:2020-01-31 22:40:17      阅读:78      评论:0      收藏:0      [点我收藏+]

标签:for   scribe   sts   class   cat   des   ssis   sample   csdn   

You are given two integers nn and mm . Calculate the number of pairs of arrays (a,b)(a,b) such that:

  • the length of both arrays is equal to mm ;
  • each element of each array is an integer between 11 and nn (inclusive);
  • aibiai≤bi for any index ii from 11 to mm ;
  • array aa is sorted in non-descending order;
  • array bb is sorted in non-ascending order.

As the result can be very large, you should print it modulo 109+7109+7 .

Input

The only line contains two integers nn and mm (1n10001≤n≤1000 , 1m101≤m≤10 ).

Output

Print one integer – the number of arrays aa and bb satisfying the conditions described above modulo 109+7109+7 .

Examples
Input
 
2 2
Output
 
5
Input
 
10 1
Output
 
55
Input
 
723 9
Output
 
157557417

由题意可知,这个题所求的两个序列实际上可以合并成一个,将递增序列的头部接到递减序列的尾部,所得一个长为2m的序列。原问题可以转化为求范围1到n的2m个数组成的递增序列的个数(一个序列里可以有两个重复的数)。官方题解直接用Python+组合数公式
from math import factorial as fact
mod = 10**9 + 7

def C(n, k):
    return fact(n) // (fact(k) * fact(n - k))

n, m = map(int, input().split())
print(C(n + 2*m - 1, 2*m) % mod)

 

从原理入手,当这2m个数里有i个数不相同时,先从n个位置里用C(n,i)算出当前有几种选法,接下来有2m-i个数是和刚刚选出来的序列的部分数相同,可以看作同球入不同盒问题直接应用公式计算即可,每次循环更新ans。题目要求取模,这里借鉴了https://www.csdn.net/gather_2b/NtzaIgxsNzQtYmxvZwO0O0OO0O0O.html里组合数快速取模的知识。

#include <bits/stdc++.h>
#define MOD 1000000007
using namespace std;
long long n,m;
int inv(int a)
{
    return a==1?1:(long long)(MOD-MOD/a)*inv(MOD%a)%MOD;
}
long long C(long long n,long long m)//组合数快速取模 
{
    if(m<0)return 0;
    if(n<m)return 0;
    if(m>n-m)m=n-m;
    long long up=1,down=1;
    long long i;
    for(i=0;i<m;i++)
    {
        up=up*(n-i)%MOD;
        down=down*(i+1)%MOD;
    }
    return up*inv(down)%MOD;
}
int main()
{
    cin>>n>>m;
    int i;
    long long ans=0; 
    for(i=2*m;i>=1;i--)//有i个不同的数 
    {
        if(i>n)continue;   
        ans=(ans+C(n,i)*/*同球入不同盒 2*m-i入i */ C(2*m-i+i-1,i-1))%MOD;
    }
    printf("%lld",ans);
    return 0;
}

 



Educational Codeforces Round 80 C. Two Arrays(组合数快速取模)

标签:for   scribe   sts   class   cat   des   ssis   sample   csdn   

原文地址:https://www.cnblogs.com/lipoicyclic/p/12246958.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!