LOJ #2721. 「NOI2018」屠龙勇士（set + exgcd）
题意
LOJ #2721. 「NOI2018」屠龙勇士
题解
首先假设每条龙都可以打死，每次拿到的剑攻击力为 \(ATK\) 。
这个需要支持每次插入一个数，查找比一个 \(\le\) 数最大的数（或者找到 \(>\) 一个数的最小数），删除一个数。
这个东西显然是可以用 std :: multiset<long long> 来处理的（手写权值线段树或者平衡树也行）。
对于每一条龙我们只能刚好一次秒杀，并且要恰好算血量最后为 \(0\)（一波带走）。
然后就转化成求很多个方程：
\[\begin{cases}
x \times ATK_1 \equiv a_1 \pmod {p_1} \\
~~~~~~~~~~~~~~~~~~~~~ \vdots \\
x \times ATK_n \equiv a_n \pmod {p_n} \\
\end{cases}
\]
求最小正整数解 \(x\) 满足这些所有方程。
如果把 \(ATK_i\) 除到右边去，也就是
\[x \equiv \frac{a_i}{ATK_i} \pmod {p_i}
\]
就转化成求模线性方程组的最小整数解了，可以参考 我的数论总结 ，但是那个板子有个地方需要 慢速乘 。
这个需要用 \(exgcd\) 计算逆元，如果没有逆元那么对于这个方程是无解的。
但是这个有点特殊情况，也就是 \(\gcd(ATK_i, a_i, p_i) > 1\) 的时候，需要约去 \(gcd\) 。
比如 \(2x \equiv 8 \pmod {36}\) 的时候，显然 \(x = 4\) 是其中的一个解，但 \(2\) 对于 \(36\) 没有逆元。
但将方程转化后 \(x \equiv 4 \pmod {18}\) 就是等价于原来方程的另一个可行方程。
然后如果这个方程仍然没有逆元的话就是真的无解了。如果合并方程组中无解那也是无解。
还有一个地方对于 \(p_i = 1\) 的情况，解出来是 \(x \equiv 0 \pmod {1}\) 。这个需要给答案有一个下界 \(a_i\) ，最后要一直加上 \(lcm\) 使得它不小于这个下界。
然后各种地方注意会爆 long long ，慢速乘就好了。（挂了 \(15pts\) 。。）
代码
#include <bits/stdc++.h>
#define For(i, l, r) for(register int i = (l), i##end = (int)(r); i <= i##end; ++i)
#define Fordown(i, r, l) for(register int i = (r), i##end = (int)(l); i >= i##end; --i)
#define Set(a, v) memset(a, v, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define debug(x) cout << #x << ": " << x << endl
#define DEBUG(...) fprintf(stderr, __VA_ARGS__)
using namespace std;
typedef long long ll;
inline ll read() {
ll x = 0, fh = 1; char ch = getchar();
for (; !isdigit(ch); ch = getchar()) if (ch == '-') fh = -1;
for (; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + (ch ^ 48);
return x * fh;
}
void File() {
#ifdef zjp_shadow
freopen ("2721.in", "r", stdin);
freopen ("2721.out", "w", stdout);
#endif
}
void Exgcd(ll a, ll b, ll &x, ll &y) {
if (!b) x = 1, y = 0;
else Exgcd(b, a % b, y, x), y -= a / b * x;
}
inline ll Mult(ll x, ll y, ll Mod) {
ll res = 0; y = (y % Mod + Mod) % Mod;
for (; y; y >>= 1, (x += x) %= Mod)
if (y & 1) (res += x) %= Mod;
return res;
}
const int N = 1e5 + 1e3;
namespace Equations {
int n; ll mod[N], rest[N];
ll Solve() {
For (i, 1, n - 1) {
ll a = mod[i], b = mod[i + 1], c = rest[i + 1] - rest[i], gcd = __gcd(a, b), k1, k2;
if (c % gcd) return - 1;
a /= gcd; b /= gcd; c /= gcd;
Exgcd(a, b, k1, k2);
k1 = Mult(k1, c, b);
mod[i + 1] = mod[i] / __gcd(mod[i], mod[i + 1]) * mod[i + 1] ;
rest[i + 1] = (mod[i] * k1 % mod[i + 1] + rest[i] % mod[i + 1] + mod[i + 1]) % mod[i + 1];
}
return rest[n];
}
void Out() {
For (i, 1, n) printf ("%lld %lld\n", mod[i], rest[i]);
}
};
multiset<ll> S;
inline ll Find(ll x) {
multiset<ll> :: iterator it = S.upper_bound(x);
if (it != S.begin()) -- it;
ll res = *it; S.erase(it); return res;
}
int n, m; ll a[N], p[N], atk[N], award[N];
inline ll Get_Inv(ll bas, ll Mod) {
if (__gcd(bas, Mod) != 1) return -1;
static ll x, y;
Exgcd (bas, Mod, x, y);
return (x % Mod + Mod) % Mod;
}
int main () {
File();
int cases = read();
while (cases --) {
n = read(); m = read();
For (i, 1, n) a[i] = read();
For (i, 1, n) p[i] = read();
For (i, 1, n)
award[i] = read();
For (i, 1, m) { ll x = read(); S.insert(x); }
For (i, 1, n)
atk[i] = Find(a[i]), S.insert(award[i]);
S.clear();
Equations :: n = n;
bool flag = true;
ll ans = 0, lcm = 1;
For (i, 1, n) {
ll gcd = __gcd(__gcd(atk[i], p[i]), a[i]);
a[i] /= gcd; atk[i] /= gcd; p[i] /= gcd;
if (p[i] == 1) {
ans = max(ans, a[i] / atk[i] + (a[i] % atk[i] ? 1 : 0));
}
ll tmp = Get_Inv(atk[i], p[i]);
if (tmp == -1) { flag = false; break; }
Equations :: mod[i] = p[i];
Equations :: rest[i] = Mult(a[i] % p[i], tmp % p[i], p[i]);
lcm = lcm / __gcd(lcm, p[i]) * p[i];
}
if (!flag) { puts("-1"); continue ; }
ll tmp = Equations :: Solve();
if (tmp == -1) { puts("-1"); continue ; }
if (tmp < ans) {
ll gap = (ans - tmp) / lcm;
tmp += gap * lcm;
while (tmp < ans) tmp += lcm;
while (tmp - lcm > ans) tmp -= lcm;
}
printf ("%lld\n", tmp);
}
#ifdef zjp_shadow
cerr << (double) clock() / CLOCKS_PER_SEC << endl;
#endif
return 0;
}
