CC QGRID [分治][树链剖分][Dijkstra]

8 篇文章 0 订阅
8 篇文章 0 订阅

Solution

%%%%%%%%%PBS
道理上和ZJOI2016旅行者是相似的。
在大佬的带领下终于打掉了这题。。
根据最短路所在分治结构中的位置来建最短路树,并操作询问答案。
自己打的时候真的失了智。刚开始 m n都读反了。。后面分治的顺序也搞乱掉了。。
代码能力好垃圾哦。。

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 601010;
const long long INF = 1ll << 60;
typedef long long ll;

inline char get(void) {
    static char buf[100000], *S = buf, *T = buf;
    if (S == T) {
        T = (S = buf) + fread(buf, 1, 100000, stdin);
        if (S == T) return EOF;
    }
    return *S++;
}
template<typename T>
inline void read(T &x) {
    static char c; x = 0;
    for (c = get(); c < '0' || c > '9'; c = get());
    for (; c >= '0' && c <= '9'; c = get()) x = x * 10 + c - '0';
}

int n, m, q, cnt;
int Step[N], H[N];
ll ans[N];
ll Ans[N];
ll x;
struct Qry {
    int opt, id;
    int i1, i2, j1, j2;
    int step, h;
    int usd;
    ll c;
    ll ans;
};
Qry a[N], b[N], d[N];
struct edge {
    int to, next;
    ll key;
    edge(int t = 0, int n = 0, ll k = 0):to(t), next(n), key(k) {}
};
int id[4][N];
template<typename T>
inline T Min(T a, T b) {
    return a < b ? a : b;
}
template<typename T>
inline T Max(T a, T b) {
    return a > b ? a : b;
}

namespace Tree {
    int Gcnt, Pcnt, clc;
    int head[N];
    int fa[N], dep[N], size[N], son[N], pos[N], top[N];
    edge G[N << 4];
    namespace BIT {
        ll c[N];
        int clc;
        int vis[N];
        int maxn;
        void Init(int n) {
            maxn = n; ++clc;
        }
        inline void Add(int x, ll a) {
            for (; x <= maxn; x += x & -x) 
                if (vis[x] == clc) c[x] += a;
                else {
                    vis[x] = clc; c[x] = a;
                }
        }
        inline ll Sum(int x) {
            ll sum = 0;
            for (; x; x -= x & -x)
                if (vis[x] == clc) sum += c[x];
            return sum;
        }
    }
    inline void AddEdge(int from, int to) {
        G[++Gcnt] = edge(to, head[from]); head[from] = Gcnt;
        G[++Gcnt] = edge(from, head[to]); head[to] = Gcnt;
    }
    void Init(int L, int R) {
        clc = Gcnt = Pcnt = 0; int i;
        for (int k = 1; k <= m; k++)
            for (int j = L; j <= R; j++) {
                i = id[k][j];
                head[i] = fa[i] = dep[i] = size[i] = son[i] = pos[i] = top[i] = 0;
        }
        BIT::Init(id[m][n]);
    }
    inline void dfs1(int u) {
        int to; size[u] = 1;
        for (int i = head[u]; i; i = G[i].next) {
            to = G[i].to; if (to == fa[u]) continue;
            fa[to] = u; dep[to] = dep[u] + 1;
            dfs1(to); size[u] += size[to];
            if (size[son[u]] < size[to]) son[u] = to;
        }
    }
    inline void dfs2(int u, int t) {
        int to; top[u] = t; pos[u] = ++Pcnt;
        if (son[u]) dfs2(son[u], t);
        for (int i = head[u]; i; i = G[i].next)
            if (G[i].to != son[u] && G[i].to != fa[u])
                dfs2(G[i].to, G[i].to);
    }
    inline void Add(int u, int v, ll c) {
        static int f1, f2;
        while ((f1 = top[u]) != (f2 = top[v])) {
            if (dep[f1] < dep[f2]) {
                swap(f1, f2); swap(u, v);
            }
            BIT::Add(pos[f1], c);
            BIT::Add(pos[u] + 1, -c);
            u = fa[f1];
        }
        if (pos[u] > pos[v]) swap(u, v);
        BIT::Add(pos[u], c);
        BIT::Add(pos[v] + 1, -c);
    }
    inline ll Query(int u) {
        return BIT::Sum(pos[u]);
    }
}
namespace Gra {
    typedef pair<ll, int> P;
    int Gcnt, st, ed, clc;
    int head[N], pre[N];
    int vis[N], able[N];
    ll dist[N];
    edge G[N << 4];
    priority_queue<P> Q;
    inline void AddEdge(int from, int to, ll key) {
        G[++Gcnt] = edge(to, head[from], key); head[from] = Gcnt;
        G[++Gcnt] = edge(from, head[to], key); head[to] = Gcnt;
    }
    void Dijkstra(int u, int L, int R, int fl = 0) {
        ++clc;
        for (int i = 1; i <= m; i++)
            for (int j = L; j <= R; j++) {
                dist[id[i][j]] = INF;
                able[id[i][j]] = clc;
                pre[id[i][j]] = vis[id[i][j]] = 0;
            }
        Gcnt = 0; int to; dist[u] = 0;
        Q.push(P(0, u));
        while (!Q.empty()) {
            int x = Q.top().second;
            Q.pop();
            if (vis[x]) continue;
            vis[x] = true;
            for (int i = head[x]; i; i = G[i].next) {
                to = G[i].to;
                if (able[to] != clc) continue;
                if (dist[to] > dist[x] + G[i].key) {
                    dist[to] = dist[x] + G[i].key;
                    if (!vis[to]) Q.push(P(-dist[to], to));
                    pre[to] = x;
                }
            }
        }
        if (fl) {
            Tree::Init(L, R);
            for (int i = 1; i <= m; i++)
                for (int j = L; j <= R; j++)
                    if (pre[id[i][j]]) Tree::AddEdge(id[i][j], pre[id[i][j]]);
            Tree::dfs1(u); Tree::dfs2(u, u);
        }
    }
}
void DivAndConq1(int L, int R, int step, int l, int r) {
    if (L > R || l > r) return;
    int Mid = (L + R) >> 1, p1, p2;
    for (int i = 1; i <= m; i++) {
        Gra::Dijkstra(id[i][Mid], L, R);
        for (int j = l; j <= r; j++)
            if (a[j].opt == 1) {
                if (ans[a[j].id] > Gra::dist[id[a[j].i1][a[j].j1]] + Gra::dist[id[a[j].i2][a[j].j2]]) {
                    ans[a[j].id] = Gra::dist[id[a[j].i1][a[j].j1]] + Gra::dist[id[a[j].i2][a[j].j2]];
                    Step[a[j].id] = step; H[a[j].id] = i;
                }
            }
    }
    p1 = l - 1; p2 = r + 1;
    for (int i = l; i <= r; i++) {
        if (a[i].opt == 1 && a[i].j1 < Mid && a[i].j2 < Mid) b[++p1] = a[i];
        if (a[i].opt == 1 && a[i].j1 > Mid && a[i].j2 > Mid) b[--p2] = a[i];
    }
    for (int i = l; i <= p1; i++) a[i] = b[i];
    for (int i = p2; i <= r; i++) a[i] = b[i];
    DivAndConq1(L, Mid - 1, step + 1, l, p1);
    DivAndConq1(Mid + 1, R, step + 1, p2, r);
}
void DivAndConq(int L, int R, int step, int l, int r) {
    if (L > R) return;
    int Mid = (L + R) >> 1, p1, p2;
    for (int i = 1; i <= m; i++) {
        Gra::Dijkstra(id[i][Mid], L, R, 1);
        for (int j = l; j <= r; j++)
            if (a[j].opt == 1) {
                if (a[j].step == step && a[j].h == i)
                    Tree::Add(id[a[j].i1][a[j].j1], id[a[j].i2][a[j].j2], a[j].c);
            } else {
                Ans[a[j].id] += Tree::Query(id[a[j].i1][a[j].j1]);
            }
    }
    p1 = l - 1; p2 = r + 1;
    for (int i = l; i <= r; i++) {
        if ((a[i].opt == 1 && a[i].j1 < Mid && a[i].j2 < Mid) || 
            (a[i].opt == 2 && a[i].j1 < Mid)) b[++p1] = a[i];
        if ((a[i].opt == 1 && a[i].j1 > Mid && a[i].j2 > Mid) || 
            (a[i].opt == 2 && a[i].j1 > Mid)) b[--p2] = a[i];
    }
    for (int i = l; i <= p1; i++) a[i] = b[i];
    for (int i = p2; i <= r; i++) a[i] = b[r - i + p2];
    DivAndConq(L, Mid - 1, step + 1, l, p1);
    DivAndConq(Mid + 1, R, step + 1, p2, r);
}

int main(void) {
    freopen("1.in", "r", stdin);
    freopen("1.out", "w", stdout);
    read(m); read(n); read(q);
    for (int j = 1; j <= n; j++)
        for (int i = 1; i <= m; i++) {
            id[i][j] = ++cnt;
            ans[id[i][j]] = INF;
        }
    for (int i = 1; i < m; i++)
        for (int j = 1; j <= n; j++) {
            read(x);
            Gra::AddEdge(id[i][j], id[i + 1][j], x);
        }
    for (int i = 1; i <= m; i++)
        for (int j = 1; j < n; j++) {
            read(x);
            Gra::AddEdge(id[i][j], id[i][j + 1], x);
        }
    for (int i = 1; i <= q; i++) {
        read(a[i].opt);
        if (a[i].opt == 1) {
            read(a[i].i1); read(a[i].j1);
            read(a[i].i2); read(a[i].j2);
            read(a[i].c); Ans[i] = -1;
        } else {
            read(a[i].i1); read(a[i].j1);
        }
        a[i].id = i; d[i] = a[i];
    }
    DivAndConq1(1, n, 1, 1, q);
    for (int i = 1; i <= q; i++) {
        a[i] = d[i];
        a[i].step = Step[i];
        a[i].h = H[i];
//      if (a[i].opt == 1 && Step[i] == 0) return cout << i << endl, 0;
    }
    DivAndConq(1, n, 1, 1, q);
    for (int i = 1; i <= q; i++)
        if (~Ans[i]) printf("%lld\n", Ans[i]);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值