【香蕉oi】小Z的城市之旅（组合数学）-CSDN博客

本文链接：https://blog.csdn.net/xyyxyyx/article/details/102473666

文章目录

题意
思路
代码

题意

在一个无限大的平面上，每次可以向上或下或左或右走一个单位长度。求从 $(0, 0)$ 走到 $(x, y)$ 共用 $r$ 步的方案数。

$T,x,y\le2\cdot10^5$ （T是数据组数）。

思路

考场上苦思冥想无果，故打表找规律。

题解思路：

把坐标轴顺时针旋转 $45\degree$ 再将单位长度缩短到 $\frac{\sqrt{2}}{2}$ ，即将 $(x, y)$ 转化为 $(x - y, x + y)$ 。

然后一次移动将会在 $x$ 轴和 $y$ 轴上同时 $\pm 1$ ，所以两个坐标轴独立，可以分开考虑，乘法原理合并。

然后问题转化为多重集排列问题， $S=\{(x+\frac{r-x}{2}) \cdot (+1),\frac{r-x}{2}\cdot (-1)\}$ ，答案为 $C_{r}^{\frac{r-x}{2}}$ 。

数数好题。~~我不会的都是好题~~。

代码

#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10, mod = 998244353;
int T, x, y, r;
 
namespace Combination
{
    int fr[N], ifr[N];
    int fpow(int x, int y){
        int ret = 1;
        while (y){
            if (y & 1) ret = 1LL * ret * x % mod;
            x = 1LL * x * x % mod;
            y >>= 1;
        }
        return ret;
    }
    void init(){
        fr[0] = 1;
        for (int i = 1; i < N; ++ i)
            fr[i] = 1LL * fr[i-1] * i % mod;
        ifr[N-1] = fpow(fr[N-1], mod-2);
        for (int i = N-2; i >= 0; -- i)
            ifr[i] = 1LL * ifr[i+1] * (i+1) % mod;
    }
    int C(int x, int y){
        return 1LL * fr[x] * ifr[y] % mod * ifr[x-y] % mod;
    }
}
using namespace Combination;
 
namespace solver1 // O(Tn)
{
    int ans;
    void main()
    {
        init();
        for (; T--; ){
            scanf("%d%d%d", &x, &y, &r);
            x = abs(x); y = abs(y);
            if ((x + y) % 2 != r % 2 || r < x + y){
                puts("0");
                continue;
            }
            ans = 0;
            for (int i = x; i <= r-y; i += 2){
                (ans += 1LL * C(r, i) * C(i, i-x>>1) % mod * C(r-i, r-i-y>>1) % mod) %= mod;
            }
            printf("%d\n", ans);
        }
    }
}
 
namespace solver2 // O(T+n)
{
    void main()
    {
        init();
        for (; T--; ){
            scanf("%d%d%d", &x, &y, &r);
            x = abs(x); y = abs(y);
            if (x < y) swap(x, y);
            if ((x + y) % 2 != r % 2 || r < x + y){
                puts("0");
                continue;
            }
            printf("%lld\n", 1LL * C(r, (r-x-y)>>1) * C(r, (r-(x-y))>>1) % mod);
        }
    }
}
 
int main()
{
    scanf("%d", &T);
    solver2::main();
    return 0;
}