hdu 1535 Invitation Cards(有向图的来回最短路，要反向建图)

题目：

链接：点击打开链接

题意：

给一个图，求1到各点和各点到1最短路。

思路：

先spfa,然后反向建图，在spfa就行了。

代码：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;
#define INF 100000000

const int N = 1000010;

struct node{
int u,v,w;
}edge
;

int dis
,vis
;
int first
,next
;
int p,m;

void spfa1(int u)
{
for(int i=0; i<=p; i++)
{
dis[i] = INF;
}
dis[u] = 0;
memset(vis,0,sizeof(vis));
queue<int> q;
q.push(u);
while(!q.empty())
{
u = q.front();
q.pop();
vis[u] = 0;
for(int k=first[u]; k!=-1; k=next[k])
{
int v = edge[k].v;
if(dis[v] > dis[u] + edge[k].w)
{
dis[v] = dis[u] + edge[k].w;
if(!vis[v])
{
vis[v] = 1;
q.push(v);
}
}
}
}
}

void spfa2(int u)
{
for(int i=0; i<=p; i++)
{
dis[i] = INF;
}
dis[u] = 0;
memset(vis,0,sizeof(vis));
queue<int> q;
q.push(u);
while(!q.empty())
{
u = q.front();
q.pop();
vis[u] = 0;
for(int k=first[u]; k!=-1; k=next[k])
{
int v = edge[k].u;
if(dis[v] > dis[u] + edge[k].w)
{
dis[v] = dis[u] + edge[k].w;
if(!vis[v])
{
vis[v] = 1;
q.push(v);
}
}
}
}
}

void buildGraph()
{
memset(next,-1,sizeof(next));
memset(first,-1,sizeof(first));
for(int i=0; i<m; i++)
{
int v = edge[i].v;
next[i] = first[v];
first[v] = i;
}
}

int main()
{
//freopen("input.txt","r",stdin);
int t;
cin>>t;
while(t--)
{
scanf("%d%d",&p,&m);
memset(first,-1,sizeof(first));
memset(next,-1,sizeof(next));
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
next[i] = first[edge[i].u];
first[edge[i].u] = i;
}
spfa1(1);
int result = 0;
for(int i=2; i<=p; i++)
{
result += dis[i];
}
buildGraph();
spfa2(1);
for(int i=2; i<=p; i++)
{
result += dis[i];
}
printf("%d\n",result);
}
return 0;
}

-----------------------------------------------------------------------

收获：

->学习到了SPFA算法：

->思想：

->模板：

void add(int i,int j,int w)
{
e[t].from=i;
e[t].to=j;
e[t].w=w;
e[t].next=head[i];
head[i]=t++;
}
void spfa(int s)
{
queue <int> q;
for(int i=1;i<=n;i++)
dist[i]=inf;
memset(vis,false,sizeof(vis));
q.push(s);
dist[s]=0;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=false;
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(dist[v]>dist[u]+e[i].w)
{
dist[v]=dist[u]+e[i].w;
if(!vis[v])
{
vis[v]=true;
q.push(v);
}
}
}
}
}

-----------------------------------------------------------

战斗，从不退缩；奋斗，永不停歇~~~~~~~~~~