Desant 2 题解
很好的题。
先不考虑区间，先想 \(l=1,r=n\) 的情况。
考虑 dp，\(f_i\) 表示考虑 \([l,r]\) 的答案。
则容易得到：
\[f_i=\max\left\{f_{i-1}, f_{i-k}+s_i-s_{i-k}\right\},f_0=0
\]
其中 \(s\) 为 \(a\) 的前缀和。
这个转移本身是 \(\Theta(n)\) 的。
遇见这种区间查询 DP 的通常都是动态 DP，但是由于下标跨了 \(k\)，矩阵的大小为 \(\Theta(k)\)，所以进行如此操作的时间复杂度为 \(\Theta(k^3\log n)\)，表现很差。
本题难点在此。
我们看到这个 DP 式子非常工整，灵光一现，唉我给她塞图上。
我们从 \(u\) 向 \(v\) 连长度为 \(w\) 的边表示 \(f_v\) 的转移方程式里有一项为 \(f_u + w\)。
突破口出现了！
原题所求的 \([l,r]\) 的答案就是 \(l-1\) 到 \(r\) 的最长路！（注意此处由于 \(l\) 也要参与运算，所以起点为 \(l-1\)）
但是新的问题到来了，怎么给这样一个图求两点最短路呢？
显然不能 Johnson，这样比暴力还慢。
显然是要利用这个图的结构。
我们经过一顿梳理发现她长这样。
规规整整的跟个矩形一样。
知道网格求最短路的同学瞬间就懂了。
接下来是个很经典的 trick。
对于一类网格图，我们若要求若干点对间的最短或长路则采用以下算法：
我们考虑把询问离线下来分治。
每次分治面向原网格中的一个子网格。
为了方便我们设她是 \(r\times c\) 的。
\(1^\circ\) 若 \(r>c\)，我们按照中间行把她切成两半，对于中间行上每个点 \(p\) 算出其对子矩阵里所有点的最短或长路。（若是 DAG 则可直接 DP），然后枚举所有被这个中间行分开的点对 \(u,v\)，将 \(u\) 到 \(v\) 的最短或长路经过 \(p\) 松弛。（因为 \(u\) 到 \(v\) 的最短或长路一定经过 \(p\)）
\(2^\circ\) 不然则按照中间列切，同样枚举中间列上的点来计算。
每次把切完的两个小网格分治下去，要把两端点都在。
每次分治 \(O(n\sqrt n)\)。
\[T(n)=O(n\sqrt n) + 2T(\cfrac{n}{2})
\]
由主定理，总复杂度 \(O(n\sqrt n)\)。
回到这道题。
我们发现还有一些从最顶上指到底下的边，他们可以通过这条边而不需通过中间行。
对于这种情况我们可以额外枚举第一行的每个点，因为经过斜边一定要经过第一行的点。
但是有可能在同一侧的节点的最长路穿下去再慢慢爬上来的情况。
这种情况我们也要给在同一侧的点对统计答案，但是由于只有一次不影响复杂度。
#pragma GCC optimize("Ofast", 3, 2, 1)
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define debug(x) cerr << #x << " = " << x << endl
#define rep(i, a, b) for (int i = (a); i <= (b); i++)
#define per(i, a, b) for (int i = (a); i >= (b); i--)
#define gn(u, v) for (int v : G.G[u])
#define pb emplace_back
#define mp make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
#define pii pair<int, int>
#define vi vector<int>
#define vpii vector<pii>
#define vvi vector<vi>
#define no cout << "NO" << endl
#define yes cout << "YES" << endl
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define tomin(x, y) ((x) = min((x), (y)))
#define tomax(x, y) ((x) = max((x), (y)))
#define ck(mask, i) (((mask) >> (i)) & 1)
#define pq priority_queue
#define FLG (cerr << "Alive!" << endl);
constexpr int MAXN = 6e5 + 5;
constexpr int INF = 0x3f3f3f3f3f3f3f3f;
int n, k, q;
int a[MAXN];
vector<vpii> G;
vector<vpii> H;
struct Query {
int u, v, ans;
};
vector<Query> query;
vi num[MAXN];
int x[MAXN], y[MAXN];
int dis[2][MAXN];
void get(int start, const vector<vpii> &G, const vector<vpii> &H, int lx, int rx, int ly, int ry, bool typ) {
// rep(i, lx, rx) {
// rep(j, ly, ry) {
// dis[typ][num[i][j]] = -INF;
// }
// }
vi vis;
if (typ) {
per(j, ry, ly) {
per(i, rx, lx) {
if (num[i][j] > start && !typ)
vis.pb(num[i][j]);
if (num[i][j] < start && typ)
vis.pb(num[i][j]);
}
}
} else {
rep(j, ly, ry) {
rep(i, lx, rx) {
if (num[i][j] > start && !typ)
vis.pb(num[i][j]);
if (num[i][j] < start && typ)
vis.pb(num[i][j]);
}
}
}
dis[typ][start] = 0;
for (int u : vis) {
if (u == start)
continue;
dis[typ][u] = -INF;
for (auto [v, w] : H[u]) {
if (x[v] < lx || x[v] > rx || y[v] < ly || y[v] > ry)
continue;
// if (u == 200)
// cerr << v << " " << w << " " << typ << endl;
if (!typ && v >= start)
tomax(dis[typ][u], dis[typ][v] + w);
if (typ && v <= start)
tomax(dis[typ][u], dis[typ][v] + w);
}
// cerr << "dis from " << start << " to " << u << " with type " << typ << " is " << dis[typ][u] << endl;
// if (u == 200) {
// cerr << dis[typ][u] << endl;
// }
}
}
void solve(int lx, int rx, int ly, int ry, vector<Query> &q) {
if (lx > rx || ly > ry || q.empty())
return;
rep(i, lx, rx) {
rep(j, ly, ry) {
dis[false][num[i][j]] = dis[true][num[i][j]] = 0;
}
}
if (rx - lx <= ry - ly) {
int mid = ly + ry >> 1;
vector<Query> l, r, now;
for (auto [u, v, ans] : q) {
if (y[u] < mid && y[v] < mid) {
l.pb(Query{u, v, ans});
} else if (y[u] > mid && y[v] > mid) {
r.pb(Query{u, v, ans});
} else {
now.pb(Query{u, v, ans});
}
}
rep(tmp, lx, rx) {
int cur = num[tmp][mid];
get(cur, G, H, lx, rx, ly, ry, false);
get(cur, H, G, lx, rx, ly, ry, true);
dis[0][cur] = 0;
dis[1][cur] = 0;
for (auto& [u, v, ans] : now) {
// if (u == 170 && v == 200) {
// cerr << cur << "." << lx << " " << rx << " " << ly << " " << ry << endl;
// cerr << dis[0][u] << " " << dis[1][v] << " " << dis[1][u] << " " << dis[0][v] << endl;
// }
tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
// cerr << dis[0][u] << " " << dis[1][u] << " " << dis[0][v] << " " << dis[1][v] << endl;
}
// for (auto& [u, v, ans] : l) {
// // if (u == 170 && v == 200) {
// // cerr << lx << " " << rx << " " << ly << " " << ry << endl;
// // cerr << dis[0][u] + dis[1][v] << " " << dis[1][u] + dis[0][v] << endl;
// // }
// tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
// // cerr << dis[0][u] << " " << dis[1][u] << " " << dis[0][v] << " " << dis[1][v] << endl;
// }
// for (auto& [u, v, ans] : r) {
// // if (u == 170 && v == 200) {
// // cerr << lx << " " << rx << " " << ly << " " << ry << endl;
// // cerr << dis[0][u] + dis[1][v] << " " << dis[1][u] + dis[0][v] << endl;
// // }
// tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
// // cerr << dis[0][u] << " " << dis[1][u] << " " << dis[0][v] << " " << dis[1][v] << endl;
// }
}
solve(lx, rx, ly, mid - 1, l);
solve(lx, rx, mid + 1, ry, r);
int i = 0, j = 0, cnt = 0;
for (auto& [u, v, ans] : q) {
if (y[u] < mid && y[v] < mid) {
ans = l[i++].ans;
} else if (y[u] > mid && y[v] > mid) {
ans = r[j++].ans;
} else {
ans = now[cnt++].ans;
}
}
} else {
int mid = lx + rx >> 1;
vector<Query> l, r, now;
for (auto [u, v, ans] : q) {
if (x[u] < mid && x[v] < mid) {
l.pb(Query{u, v, ans});
} else if (x[u] > mid && x[v] > mid) {
r.pb(Query{u, v, ans});
} else {
now.pb(Query{u, v, ans});
}
}
rep(tmp, ly, ry) {
int cur = num[mid][tmp];
get(cur, G, H, lx, rx, ly, ry, false);
get(cur, H, G, lx, rx, ly, ry, true);
dis[0][cur] = 0;
dis[1][cur] = 0;
for (auto& [u, v, ans] : now) {
// if (u == 170 && v == 200) {
// cerr << cur << " " << lx << " " << rx << " " << ly << " " << ry << endl;
// cerr << dis[0][u] << " " << dis[1][v] << " " << dis[1][u] << " " << dis[0][v] << endl;
// }
tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
}
// for (auto& [u, v, ans] : l) {
// tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
// }
// for (auto& [u, v, ans] : r) {
// tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
// }
}
if (lx == 1 && rx == k) {
rep(tmp, ly, ry) {
int cur = num[lx][tmp];
get(cur, G, H, lx, rx, ly, ry, false);
get(cur, H, G, lx, rx, ly, ry, true);
dis[0][cur] = 0;
dis[1][cur] = 0;
for (auto& [u, v, ans] : now) {
// if (u == 170 && v == 200) {
// cerr << cur << " " << lx << " " << rx << " " << ly << " " << ry << endl;
// cerr << dis[0][u] << " " << dis[1][v] << " " << dis[1][u] << " " << dis[0][v] << endl;
// }
tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
}
for (auto& [u, v, ans] : l) {
tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
}
for (auto& [u, v, ans] : r) {
tomax(ans, max(dis[0][u] + dis[1][v], dis[1][u] + dis[0][v]));
}
}
}
solve(lx, mid - 1, ly, ry, l);
solve(mid + 1, rx, ly, ry, r);
int i = 0, j = 0, cnt = 0;
for (auto& [u, v, ans] : q) {
if (x[u] < mid && x[v] < mid) {
tomax(ans, l[i++].ans);
} else if (x[u] > mid && x[v] > mid) {
tomax(ans, r[j++].ans);
} else {
tomax(ans, now[cnt++].ans);
}
}
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n >> k >> q;
int tmp = (n / k + 1) * k - 1;
rep(i, 1, n)
cin >> a[i];
rep(i, 1, tmp)
a[i] += a[i - 1];
G.resize(tmp + 1);
H.resize(tmp + 1);
rep(i, 1, tmp) {
G[i - 1].pb(mp(i, 0));
H[i].pb(mp(i - 1, 0));
if (i >= k)
G[i - k].pb(mp(i, a[i] - a[i - k])),
H[i].pb(mp(i - k, a[i] - a[i - k]));
}
query.resize(q);
rep(i, 0, q - 1) {
cin >> query[i].u >> query[i].v;
query[i].u--;
query[i].ans = 0;
}
rep(i, 1, k) {
num[i].resize(n / k + 2);
}
rep(i, 0, (n / k + 1) * k - 1) {
num[i % k + 1][i / k + 1] = i;
x[i] = i % k + 1;
y[i] = i / k + 1;
}
// rep (i, 0, n) {
// for (auto [v, w] : G[i]) {
// cerr << i << " " << v << " " << w << endl;
// }
// }
solve(1, k, 1, n / k + 1, query);
// cerr << x[170] << " " << y[170] << " " << x[200] << " " << y[200] << endl;
rep(i, 0, q - 1)
cout << query[i].ans << endl;
return 0;
}
