2025“钉耙编程”中国大学生算法设计暑期联赛（3） 1003 光线折射题解记录

原创于 2025-08-19 16:51:22 发布 · 882 阅读

CC 4.0 BY-SA版权

文章标签：

大致题意：在二维坐标系中，每个整点位置都有一垂直于 $y = - x$ 的玻璃，光线可折射和反射。折射率为 $a$ ，反射率为 $b$ 。现从 $(- 0.5, 0)$ 朝 $x$ 轴正方向射出光强为 $1$ 的光线，问 $(n + 0.5, m)$ 处的光强。
解：
令 $x = n + 1, y = m$ ，期间共经过了 $x + y$ 次玻璃。假设与玻璃接触后射向 $x$ 轴正方向记为 $0$ ，射向 $y$ 轴正方向记为 $1$ ，那么就可以得到一个以 $0$ 结尾， $x$ 个 $0$ 和 $y$ 个 $1$ 组成的 $01$ 字符串。其中，相邻两个字符相同会贡献 $×a\times a$ ，否则贡献 $×b\times b$ 。
现转化为：所有可能的字符串权值之和。
假设有 $k_1$ 个 $1$ 段， $k_0$ 个 $0$ 段。
当起始位置为 $1$ 的时候，形如 $1…01\dots 0$ ，此时 $k_0=k_1$ 。另， $x$ 个 $0$ 分成 $k_0$ 段，会出现 $x-k_0$ 个 $00$ 。所以此时权值为： $ax−k1+y−k1×b2k1−1a^{x-k_1+y-k_1}\times b^{2k_1-1}$ ，除此，由于是朝 $x$ 轴正方向发射，故还会多出一个 $01$ ，即： $ax+y−2k1×b2k1a^{x+y-2k_1}\times b^{2k_1}$ 。
当起始位置为 $0$ 的时候，形如 $0…00\dots 0$ ，此时 $k_0=k_1+1$ 。算上初始发射时候的 $00$ ，权值为同样为 $ax+y−2k1×b2k1a^{x+y-2k_1}\times b^{2k_1}$ 。
从这里，我们可以开始考虑枚举 $k_1$ 。
如果起始位置为 $1$ ，总的答案为： $Cx−1k1−1×Cy−1k1−1×ax+y−2k1×b2k1C_{x-1}^{k_1-1}\times C_{y-1}^{k_1-1}\times a^{x+y-2k_1}\times b^{2k_1}$ 。因为 $x$ 个 $0$ 有 $x - 1$ 个空位，从这些空位选取 $k_1-1$ 个截断会得到 $k_1$ 个段。
如果起始位置为 $0$ ，那么答案就是 $Cx−1k1×Cy−1k1−1×ax+y−2k1×b2k1C_{x-1}^{k_1}\times C_{y-1}^{k_1-1}\times a^{x+y-2k_1}\times b^{2k_1}$ 。
二者加起来得到最终式子： $Cxk1×Cy−1k1−1×ax+y−2k1×b2k1C_{x}^{k_1}\times C_{y-1}^{k_1-1}\times a^{x+y-2k_1}\times b^{2k_1}$ 。接下来就枚举 $k∈[1,min(x,y)]k\in[1,min(x,y)]$ 。
此时有一个需要注意的，如果每次都 $l o g$ 计算会 $tl e$ 。需要预处理出 $f a c$ 和 $in v f a c$ 及 $in v$ 。同时：记 $fi=Cxi×Cy−1i−1×ax+y−2i×b2if_i=C_{x}^{i}\times C_{y-1}^{i-1}\times a^{x+y-2i}\times b^{2i}$ ，那么 $fi+1=fi×(y−i)×(x−i)×inv(i)×inv(i+1)f_{i+1}=f_i\times (y-i)\times (x-i)\times inv(i)\times inv(i+1)$ ，用此式子递推可以避免每次都暴力用快速幂计算 $a^{x+y-2i}$ 和 $b^{2i}$ 。

#include<bits/stdc++.h>

#define endl '\n'
#define pii pair<int,int>
#define int long long
#define uint unsigned long long

using namespace std;

const double eps = 1e-10;
const int mod = 1e9 + 7;
const int inf = 1e18;

mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());

inline int sub(int a, int b) {
    a -= b;
    a %= mod;
    if (a < 0) return a + mod;
    else return a;
}

inline int add(int a, int b) {
    return (a + b) % mod;
}

inline int mul(int a, int b) {
    return a * b % mod;
}

inline int lowbit(int x) {
    return x & (-x);
}

inline int get(int x, int pos) {
    return (x >> pos) & 1;
}


inline int qp(int a, int b, int p) {
    if (b == 0) return 1;
    if (a == 0) return 0;
    int res = 1;
    while (b) {
        if (b & 1) {
            res = res % p * (a % p) % p;
        }
        b >>= 1;
        a = a % p * (a % p) % p;
    }
    return res % p;
}

inline uint rdm() {
    return rng();
}

const int N = 63;
int dx[] = {0, 1, -1, 0};
int dy[] = {1, 0, 0, -1};


const int NN = 1000000;
vector<int> fac(NN + 4);
vector<int> invfac(NN + 4);
vector<int> inv(NN + 4);

void init() {
    fac[0] = 1;
    for (int i = 1; i <= NN; i++) {
        fac[i] = mul(fac[i - 1], i);
        inv[i] = qp(i, mod - 2, mod);
    }
    invfac[NN] = qp(fac[NN], mod - 2, mod);
    for (int i = NN - 1; i >= 0; i--) {
        invfac[i] = mul(invfac[i + 1], i + 1);
    }
}

int cnm(int n, int m) {
    if (m < 0 || m > n) return 0;
    return mul(fac[n], mul(invfac[m], invfac[n - m]));
}

void solve() {
    int n, m, c, d;
    cin >> n >> m >> c >> d;
    int x = n + 1, y = m;
    int ans = 0;
    int t = qp(c + d, mod - 2, mod);
    int a = mul(c, t), b = mul(d, t);
    int inva = qp(a, mod - 2, mod);

    if (y == 0) {
        cout << qp(a, x, mod) << endl;
        return;
    }

    if (b == 0) {
        cout << 0 << endl;
        return;
    }

    if (a == 0) {
        if (x != y) cout << 0 << endl;
        else cout << 1 << endl;
        return;
    }

    int ab = mul(mul(b, b), mul(inva, inva));
    int l = mul(x, mul(qp(a, x + y - 2, mod), qp(b, 2, mod)));
    for (int i = 1; i <= min(x, y); i++) {
        ans = add(ans, l);
        l = mul(l, sub(x, i));
        l = mul(l, sub(y, i));
        l = mul(l, mul(inv[i], inv[i + 1]));
        l = mul(l, ab);
    }

    cout << ans << endl;

}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T = 1;
    cin >> T;
    init();
    while (T--) {
        solve();
    }
}