POJ-1511 Invitation Cards

题目链接：https://vjudge.net/problem/POJ-1511

一道简单的最短路变形，正向图和反向图各搜一遍加起来就OK，要用spfa或者dijkstra+heap

但是数据量比较大，所以卡了挺久，一是范围要开long long，不然会WA

二是以前习惯了用vector做邻接表，但这道题一直卡着超时，用数组做了邻接表才AC

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#include<queue>using namespace std;const int N=1000000+10;const long long inf=1e10+10;struct edge{int to;int c;int next;};long long vis[N],vis0[N];bool d[N];edge g[N],g0[N];int head[N],head0[N];int cnt,cnt0;int t,n,m;int main(){scanf("%d",&t);while(t--){scanf("%d%d",&n,&m);int a,b,c;memset(vis,inf,sizeof(vis));memset(vis0,inf,sizeof(vis0));memset(head,-1,sizeof(head));memset(head0,-1,sizeof(head0));for(int i=1;i<=m;i++){scanf("%d%d%d",&a,&b,&c);g[i].to=b;g[i].c=c;g[i].next=head[a];head[a]=i;g0[i].to=a;g0[i].c=c;g0[i].next=head0[b];head0[b]=i;}queue<int> q;memset(d,false,sizeof(d));q.push(1);vis[1]=0;while(!q.empty()){int pos=q.front();q.pop();d[pos]=false;for(int i=head[pos];i!=-1;i=g[i].next){int v=g[i].to;int c=g[i].c;if(vis[v]>vis[pos]+c){vis[v]=vis[pos]+c;if(!d[v])q.push(v);d[v]=true;}}}q.push(1);memset(d,false,sizeof(d));vis0[1]=0;while(!q.empty()){int pos=q.front();q.pop();d[pos]=false;for(int i=head0[pos];i!=-1;i=g0[i].next){int v=g0[i].to;int c=g0[i].c;if(vis0[v]>vis0[pos]+c){vis0[v]=vis0[pos]+c;if(!d[v])q.push(v);d[v]=true;}}}long long ans=0;for(int i=1;i<=n;i++)ans+=vis[i]+vis0[i];printf("%lld\n",ans);}return 0;}