B - 区间的连续段
思路:
可以先预处理出每个左端点能向右最远扩展到的位置, 这样就是固定左端点分成一段的最大长度, 然后可以用倍增
ST
S
T
表记录固定左端点,分成
2k
2
k
段
k=0,1,2,3.....
k
=
0
,
1
,
2
,
3.....
向右扩展到最远的位置, 查询的时候就可以
O(log n)
O
(
l
o
g
n
)
查询了
#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 1e6 + 10;
using namespace std;
int n, m, q;
int st[maxn][21];
ll a[maxn], k;
int query(int l, int r) {
int ans = 0;
for(int i = 20; i >= 0; i--) {
int nxt = st[l][i];
if(nxt == -1 || nxt > r) continue;
ans += (1 << i); l = nxt + 1;
}
if(l > r || ~st[l][0]) return ans + (l <= r);
else return -1;
}
int main() {
while(scanf("%d %d %lld", &n, &q, &k) != EOF) {
ll sum = 0; int id = 1;
for(int i = 1; i <= n; i++) scanf("%lld", &a[i]);
for(int i = 1; i <= n; i++) {
id = max(id, i);
while(sum + a[id] <= k && id <= n) {
sum += a[id]; id++;
}
if(sum) sum -= a[i];
if(id == i) st[i][0] = -1;
else st[i][0] = id - 1;
}
st[n + 1][0] = -1;
for(int i = 1; i <= 20; i++) {
st[n + 1][i] = -1;
for(int j = 1; j <= n; j++) {
int r1 = st[j][i - 1];
if(r1 == -1) st[j][i] = -1;
else st[j][i] = st[r1 + 1][i - 1];
}
}
while(q--) {
int l, r; scanf("%d %d", &l, &r);
int ans = query(l, r);
if(~ans) cout << ans << endl;
else cout << "Chtholly" << endl;
}
}
return 0;
}
E - 无向图中的最短距离
思路:
先
bfs
b
f
s
一遍, 记录每两个点之间的距离,可以知道两个点之间的距离不会超过
n
n
, 然后可以用压位的思想,记:将距离第
i
i
这个点距离为的第
j
j
个位置为
1
1
,最后求一个前缀并,那么在查询时可以
O(1)
O
(
1
)
知道答案, 对所有的查询点对的结果并起来,
bit
b
i
t
位为
1
1
的个数就是答案了
#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 1e3 + 10;
using namespace std;
int n, m, q, T, kase = 1;
vector<int> G[maxn];
int dis[maxn][maxn];
bitset<maxn> bit[maxn][maxn];
void bfs(int s) {
dis[s][s] = 0;
queue<int> que; que.push(s);
while(!que.empty()) {
int u = que.front(); que.pop();
for(int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
if(~dis[s][v]) continue;
dis[s][v] = dis[s][u] + 1; que.push(v);
}
}
}
int main() {
while(scanf("%d %d %d", &n, &m, &q) != EOF) {
memset(dis, -1, sizeof dis);
for(int i = 0; i < maxn; i++) G[i].clear();
for(int i = 0; i < m; i++) {
int u, v; scanf("%d %d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
for(int i = 1; i <= n; i++) {
bfs(i);
for(int j = 1; j <= n; j++) bit[i][dis[i][j]][j] = 1;
for(int j = 1; j <= n; j++) bit[i][j] |= bit[i][j - 1];
}
while(q--) {
int a, u, v;
scanf("%d", &a);
bitset<maxn> res;
while(a--) {
scanf("%d %d", &u, &v);
res |= bit[u][v];
}
int ans = res.count();
printf("%d\n", ans);
}
}
return 0;
}