标签：tmp maxk int 短路 2019CCPC 权值 path 1004 multiset

题意：给定一个带权有向图，有q组询问，每次询问在有向图的所有路径中，第k小的路径权值

解题思路：因为k最大只有5e4，考虑暴力搜索出前maxk小的路径并用数组记录权值，然后就可以O(1)查询。

具体实现：暴力搜索时可以借助Dijkstra最短路的思想，即用已知的最短路更新得出新的最短路。先将所有的边都装进一个multiset里面，然后每次将multiset里的首元素取出，作为新的答案，然后再用它来更新新的最短路，这样不断扩散的话就可以得到答案。

但是，这样可能会TLE或MLE，考虑再加加优化，首先我们只需要前maxk小的路径，所以multiset的可以限制在maxk以内，这样就不会MLE了，然后我们还可以先对每个节点的邻接表中的边按权值从小到大排序，这样在枚举的时候如果新路径的权值大于multiset中的最大值就可以直接break掉，这样就不会TLE了。

AC代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const int maxn=5e4+5;
const int MAXK=5e4+5;
int n,m,q,k;

struct Edge{
    int id,u,v;
    ll w;
    bool operator<(const Edge& b)const{return w<b.w;}
}; 

multiset<Edge>s;

vector<Edge>G[maxn];
ll ans[MAXK];
int tot,maxk,qry[maxn];

void init(){
    for(int i=1;i<=n;i++)sort(G[i].begin(),G[i].end());
    int cnt=0;
    while(true){
        
        //cout<<"S:\n";for(auto i:s)cout<<i.u<<" "<<i.v<<" "<<i.w<<"\n";
        
        
        ans[++cnt]=s.begin()->w;
        Edge e=*s.begin(),tmp;
        s.erase(s.begin());
        if(cnt>=maxk)break;
        
        int psz=(int)s.size();
        
        int sz=(int)G[e.v].size();
        
        if((int)s.size()+sz<=maxk){
            for(int i=0;i<sz;i++){
                Edge t=G[e.v][i];
                tmp.id=++tot;tmp.u=e.u;tmp.v=t.v;tmp.w=e.w+t.w;
                s.insert(tmp);
            }
        }
        else{
            for(int i=0;i<sz;i++){
                Edge t=G[e.v][i];
                if((int)s.size()<=maxk){
                    tmp.id=++tot;tmp.u=e.u;tmp.v=t.v;tmp.w=e.w+t.w;
                    s.insert(tmp);
                }
                else{
                    Edge last=*(--s.end());
                    if(last.w>e.w+t.w){
                        s.erase(last);
                        tmp.id=++tot;tmp.u=e.u;tmp.v=t.v;tmp.w=e.w+t.w;
                        s.insert(tmp);
                    }
                    else break;
                }
            }
        }
    }
}

int main()
{
//#ifndef ONLINE_JUDGE
//  freopen("in.txt","r",stdin);
//#endif
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d %d %d",&n,&m,&q);
        
        for(int i=1;i<=n;i++)G[i].clear();
        s.clear();tot=0;
        
        int u,v;
        ll w;
        Edge tmp;
        for(int i=1;i<=m;i++){
            scanf("%d %d %lld",&u,&v,&w);
            tmp.id=++tot;tmp.u=u;tmp.v=v;tmp.w=w;
            G[u].push_back(tmp);
            s.insert(tmp);
        }
        maxk=0;
        for(int i=1;i<=m;i++){
            scanf("%d",&qry[i]);
            maxk=max(maxk,qry[i]);
        }
        init();
        for(int i=1;i<=m;i++){
            printf("%lld\n",ans[qry[i]]);
        }
    }
    return 0;
}

标签：tmp,maxk,int,短路,2019CCPC,权值,path,1004,multiset
来源： https://www.cnblogs.com/zengzk/p/11403194.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。