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;
}
上一篇:「bzoj1925」「Sdoi2010」地精部落 (计数型dp)


下一篇:numpy中 array数组的shape属性