牛客练习赛14

B - 区间的连续段

$\ $

思路：

$\ \ \ \ \ $可以先预处理出每个左端点能向右最远扩展到的位置， 这样就是固定左端点分成一段的最大长度， 然后可以用倍增$ST$表记录固定左端点，分成$2^{k}$段$k = 0, 1, 2, 3.....$向右扩展到最远的位置， 查询的时候就可以$O(log\ 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$一遍， 记录每两个点之间的距离，可以知道两个点之间的距离不会超过$n$， 然后可以用压位的思想，记$bit[i][dis[i][j]][j]$：将距离第$i$这个点距离为$dis[i][j]$的第$j$个$bit$位置为$1$，最后求一个前缀并，那么在查询$(x_{i}, y_{i})$时可以$O(1)$知道答案， 对所有的查询点对的结果并起来， $bit$位为$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;
}