The Shortest Statement CodeForces - 1051F LCA+最短路

43 篇文章 0 订阅

太弱了。。。

一开始看到题感觉是跑一个最小生成树在上边进行LCA就行了,但是发现过不了样例,然后就是就想到了之前做过类似做法的题目,就是非生成树上的边最多只有21条,然后就那些边记录下来,通过每一条边的两个点跑lca然后取最小值

 ll ans=lca(x,y);
 for(int i=1;i<=k;i++)           
 {
       ans=min(ans,b[i].val+lca(x,b[i].x)+lca(y,b[i].y));
       ans=min(ans,b[i].val+lca(x,b[i].y)+lca(y,b[i].x));
 }
       cout<<ans<<endl;

这样发现也不行,想明白了之后你要求的可能是某两个点由多条非生成树边连接并且最短,emmm然后就没办法做了,没有想到转化成为这些非生成树边的点(最多42个)每一个点到两个询问点的距离,其实也就是跑40多次的最短路,用dij就可以卡过去。

我的做法就是先用kuske建立一个生成树(好像没有必要用kuske提出来最短边),跑dfs用倍增提前记录好父亲节点以备跑lca,然后记录其他没有在生成树里的边上的点,然后从这些点出发跑最短路,以Dis[i][x]记录第i个点到x的最短距离,然后询问x,y的最短路,就是将lca(x,y)与每一个Dis[i][x]+Dis[i][y]进行比较取最小值,即可得出答案。

代码如下

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
#define pli pair<long long,int >
const int dir[4][2]= {0,-1,-1,0,0,1,1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=2e5+5;
int f[maxn][30],d[maxn],head[maxn];
int fa[maxn];
ll dis[maxn],Dis[50][maxn];
int vis[maxn],mm[50];
int sign,n,m,q,cnt;
map<ll,int>mp;
struct node
{
    int to,p;
    ll val;
}edge[maxn<<1];
struct nod
{
    int x,y;
    ll val;
}a[maxn];
struct eg {
    ll val;
    int next;
    bool friend operator<(eg q,eg e)
    {
        return q.val>e.val;
    }
};
vector<eg>v[maxn];
bool cmp(nod s,nod e)
{
    return s.val<e.val;
}
void add(int u,int v,ll val)
{
    edge[sign]=node{v,head[u],val};
    head[u]=sign++;
}
int find(int x)
{
    if(x==fa[x]) return x;
    else return fa[x]=find(fa[x]);
}
void init()
{
    sign=cnt=0;
    mp.clear();
    for(int i=0;i<=n;i++){
        fa[i]=i;
        head[i]=-1;
        v[i].clear();
    }
}
void kuske()
{
    for(int i=1;i<=m;i++)
    {
        int fx=find(a[i].x);
        int fy=find(a[i].y);
        int x=a[i].x,y=a[i].y;
        ll val=a[i].val;
        if(fx!=fy)
        {
            fa[fx]=fy;
            add(a[i].x,a[i].y,a[i].val);//这是取出生成树的边跑倍增,其实可以直接dfs跑出来
            add(a[i].y,a[i].x,a[i].val);
        } else
        {
            if(mp[x]==0) mp[x]=++cnt,mm[cnt]=x;//记录那些点的顺序
            if(mp[y]==0) mp[y]=++cnt,mm[cnt]=y;
        }
        v[x].push_back(eg{val,y});//所有的边都加到里边跑dij
        v[y].push_back(eg{val,x});
    }
}
void dij(int id,int s)//跑出来Dis[i][x]第i点到每个点的最小距离
{
    for(int i=0;i<maxn;i++)
    {
        vis[i]=0;
        Dis[id][i]=inFF;
    }
    Dis[id][s]=0;
    priority_queue<eg >q;
    q.push(eg{0,s});
    while(!q.empty())
    {
        eg now=q.top();
        q.pop();
        int u=now.next;
        if(now.val>Dis[id][u]) continue;
        vis[u]=1;
        //cout<<mm[id]<<"->"<<now.next<<" "<<Dis[id][u]<<endl;
        for(int i=0;i<v[u].size();i++)
        {
            int next=v[u][i].next;
            ll val=v[u][i].val;
            if(vis[next]) continue;
            if(Dis[id][next]>Dis[id][u]+val)
            {
                Dis[id][next]=Dis[id][u]+val;
                q.push(eg{Dis[id][next],next});
            }
        }
    }
}
void dfs(int u)
{
    for(int i=1;(1<<i)<=n;i++)
        f[u][i]=f[f[u][i-1]][i-1];
    for(int i=head[u];~i;i=edge[i].p)
    {
        int v=edge[i].to;
        if(v==f[u][0]) continue;
        d[v]=d[u]+1;
        f[v][0]=u;
        dis[v]=dis[u]+edge[i].val;
        dfs(v);
    }
}
ll lca(int a,int b)//lca模板
{
    int s=a,e=b;
    if(d[a]<d[b]) swap(a,b);
    int x=d[a]-d[b];
    for(int i=0;(1<<i)<=x;i++)
        if((1<<i)&x) a=f[a][i];
    if(a!=b)
    {
        for(int i=(int)log2(n);i>=0;i--)
            if(f[a][i]!=f[b][i]) a=f[a][i],b=f[b][i];
        a=f[a][0];
    }
    ll ans=dis[s]+dis[e]-2*dis[a];
    return ans;
}
int main()
{
    int x,y;
    while(cin>>n>>m)
    {
        init();
        for(int i=1;i<=m;i++)
            scanf("%d %d %lld",&a[i].x,&a[i].y,&a[i].val);
        sort(a+1,a+1+m,cmp);
        kuske();
        for(int i=1;i<=cnt;i++)
            dij(i,mm[i]);
        f[1][0]=1,d[1]=0,dis[1]=0;
        dfs(1);
        cin>>q;
        while(q--)
        {
            scanf("%d %d",&x,&y);
            ll ans=lca(x,y);
            for(int i=1;i<=cnt;i++)
                ans=min(ans,Dis[i][x]+Dis[i][y]);
            printf("%lld\n",ans);
        }
    }
    return 0;
}

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值