The Shortest Statement CodeForces - 1051F（待测试）

#include <iostream>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cmath>
#define rap(i, a, n) for(int i=a; i<=n; i++)
#define MOD 2018
#define LL long long
#define ULL unsigned long long
#define Pair pair<int, int>
#define mem(a, b) memset(a, b, sizeof(a))
#define _  ios_base::sync_with_stdio(0),cin.tie(0)
//freopen("1.txt", "r", stdin);
using namespace std;
const int maxn =  * 1e5 + , INF = 0x7fffffff;
LL n, m, cnt1, cnt2, c;
LL head1[maxn], head2[maxn], f[maxn], vis[maxn], bz[maxn];
LL dis[][maxn], d[maxn];
LL res[maxn];
vector<LL> se;
struct node
{
    LL u, v, w, next;
}Node[maxn];
 
struct edge
{
    LL u, v, id, next;
}Edge[maxn];
 
void add1_(LL u, LL v, LL w)
{
    Node[cnt1].u = u;
    Node[cnt1].v = v;
    Node[cnt1].w = w;
    Node[cnt1].next = head1[u];
    head1[u] = cnt1++;
}
 
void add1(LL u, LL v, LL w)
{
    add1_(u, v, w);
    add1_(v, u, w);
}
 
void add2_(LL u, LL v, LL id)
{
    Edge[cnt2].u = u;
    Edge[cnt2].v = v;
    Edge[cnt2].id = id;
    Edge[cnt2].next = head2[u];
    head2[u] = cnt2++;
}
 
void add2(LL u, LL v, LL id)
{
    add2_(u, v, id);
    add2_(v, u, id);
}
 
LL find(LL x)
{
    return f[x]==x?x:(f[x] = find(f[x]));
}
 
LL lca(LL u, LL deep, LL root)
{
    f[u] = u;
    d[u] = deep;
    vis[u] = root;  // 标记属于的树
    for(LL i=head1[u]; i!=-; i=Node[i].next)
    {
        node e = Node[i];
        if(vis[e.v] == -)
        {
            bz[i/] = ;
            lca(e.v, deep+e.w, root);
            f[e.v] = u;
        }
    }
    for(LL i=head2[u]; i!=-; i=Edge[i].next)
    {
        edge e = Edge[i];
        if(vis[e.v] == root)   //判断另一个结点是不是和u属于一个树
        {
            LL k = find(e.v);  //寻找最近公共祖先
            res[e.id] = d[u] + d[e.v] - *d[k];
        }
    }
}
 
void spfa(LL s, LL id)
{
    for(LL i=; i<=n; i++) dis[id][i] = INF;
    mem(vis, );
    queue<LL> Q;
    Q.push(s);
    vis[s] = ;
    dis[id][s] = ;
    while(!Q.empty())
    {
        LL u = Q.front(); Q.pop();
        vis[u] = ;
        for(LL i=head1[u]; i!=-; i=Node[i].next)
        {
            node e = Node[i];
            if(dis[id][e.v] > dis[id][u] + e.w)
            {
                dis[id][e.v] = dis[id][u] + e.w;
                if(!vis[e.v])
                {
                    Q.push(e.v);
                    vis[e.v] = ;
                }
            }
        }
    }
}
 
void init()
{
    mem(head1, -);
    mem(head2, -);
    mem(res, -);
    mem(vis, -);
    cnt1 = cnt2 = ;
}
 
int main()
{
    cin >> n >> m;
    init();
    rap(i, , m)
    {
        LL u, v, w;
        scanf("%lld%lld%lld", &u, &v, &w);
        add1(u, v, w);
    }
    cin >> c;
    for(LL i=; i<=c; i++)
    {
        LL u, v;
        scanf("%lld%lld", &u, &v);
        add2(u, v, i);
    }
    for(LL i=; i<=n; i++)
        if(vis[i] == -)
            lca(i, , i);
    mem(vis, -);
    for(LL i=; i<cnt1; i++)
    {
        if(!bz[i/])
        {
            if(vis[Node[i].u] == -) se.push_back(Node[i].u), vis[Node[i].u] = ;
            if(vis[Node[i].v] == -) se.push_back(Node[i].v), vis[Node[i].v] = ;
        }
    }
    for(LL i=; i<se.size(); i++)
    {
      //  cout << se[i] << endl;
        spfa(se[i], i);
 
    }
 
    for(LL i=; i<=c; i++)
    {
        for(LL j=; j<se.size(); j++)
            res[i] = min(res[i], dis[j][Edge[i*-].u] + dis[j][Edge[i*-].v]);
        printf("%lld\n", res[i]);
    }
 
    return ;
}

The Shortest Statement

CodeForces - 1051F