AoCoder 1983 [AGC001E] BBQ Hard（组合数+dp）

最新推荐文章于 2023-06-16 18:37:04 发布

ikrvxt

最新推荐文章于 2023-06-16 18:37:04 发布

阅读量456

点赞数 1

分类专栏： DP # 排列组合文章标签：动态规划算法

本文链接：https://blog.csdn.net/Emm_Titan/article/details/122597676

版权

DP 同时被 2 个专栏收录

72 篇文章 3 订阅

订阅专栏

排列组合

20 篇文章 0 订阅

订阅专栏

problem

洛谷链接

solution

$\sum_{i=1}^{n}\sum_{j=i+1}^n\binom{a_i+b_i+a_j+b_j}{a_i+a_j}=\frac{\sum_{i=1}^{n}\sum_{j=1}^n\binom{a_i+b_i+a_j+b_j}{a_i+a_j}-\sum_{i=1}^{n}\binom{2(a_i+b_i)}{2a_i}}{2}$

$C_{x+y}^x$ 可以视为网格图中 $(0,0)\rightarrow (x,y)$ 的路径方案数。

则 $C_{a_i+b_i+a_j+b_j}^{a_i+a_j}=C_{(a_i+b_i)-(-a_j,-b_j)}^{a_i-(-a_j)}$ 即视为 $(-a_j,-b_j)\rightarrow (a_i,b_i)$ 的路径方案数。

那么这道题就是多源汇路径数了。

在一开始读入数据的时候，起点方案数增加一， $f[-a_i][-b_i]++$ 。

然后暴力转移 $d p$ ， $f_{i,j}\leftarrow^+ f_{i-1,j}+f_{i,j-1}$ ，求出到 $f[a_i][b_i]$ 的方案数。

网格图的坐标 $\in[-2000,2000]$ ，整体平移 $2 e 3$ ，这都是细节实现。

code

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define mod 1000000007
#define MAX 2000
int a[200005], b[200005];
int fac[8005], inv[8005];
int f[4005][4005];
int n;

int qkpow( int x, int y ) {
    int ans = 1;
    while( y ) {
        if( y & 1 ) ans = ans * x % mod;
        x = x * x % mod;
        y >>= 1;
    }
    return ans;
}

int C( int n, int m ) { 
    if( n < m ) return 0;
    else return fac[n] * inv[m] % mod * inv[n - m] % mod;
}

signed main() {
    scanf( "%lld", &n );
    for( int i = 1;i <= n;i ++ ) {
        scanf( "%lld %lld", &a[i], &b[i] );
        f[MAX - a[i]][MAX - b[i]] ++;
    }
    fac[0] = inv[0] = 1; 
    for( int i = 1;i <= (MAX << 2);i ++ ) fac[i] = fac[i - 1] * i % mod;
    inv[MAX << 2] = qkpow( fac[MAX << 2], mod - 2 );
    for( int i = (MAX << 2) - 1;i;i -- ) inv[i] = inv[i + 1] * ( i + 1 ) % mod;
    for( int i = 0;i <= (MAX << 1);i ++ )
        for( int j = 0;j <= (MAX << 1);j ++ ) {
            if( i ) ( f[i][j] += f[i - 1][j] ) %= mod;
            if( j ) ( f[i][j] += f[i][j - 1] ) %= mod;
        }
    int ans = 0, cnt = 0;
    for( int i = 1;i <= n;i ++ ) {
        ( cnt += C( a[i] + b[i] << 1, a[i] << 1 ) ) %= mod;
        ( ans += f[MAX + a[i]][MAX + b[i]] ) %= mod;
    }
    printf( "%lld\n", ( ans - cnt + mod ) % mod * qkpow( 2, mod - 2 ) % mod );
    return 0;
}