POJ1511,Invitation Cards(Dijkstra)
直接用Dijkstra就行了,要用优先队列优化。AC代码,Time:6000MS(在网上搜大佬的代码改的,感觉与我原先的基本差不多):#include<cstdio>#include<iostream>#include<algorithm>#include<cstring>#include<cmath>#include<...
·
直接用Dijkstra就行了,要用优先队列优化。
AC代码,Time:6000MS(在网上搜大佬的代码改的,感觉与我原先的基本差不多):
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#define inf 1000000010
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int u[maxn],v[maxn],d[maxn];
int dis[maxn];
bool vis[maxn];
struct edge
{
int v;
int dist;
bool operator < (const edge& a) const
{
return dist>a.dist;
}
edge(int _v=0,int _dist=0):v(_v),dist(_dist){}
};
vector<edge> G[maxn];
ll dijkstra(int n)
{
priority_queue<edge> Q;
for(int i=1;i<=n;i++)
{
dis[i]=inf;
vis[i]=0;
}
dis[1]=0;
Q.push(edge(1,0));
while(!Q.empty())
{
edge top=Q.top(); Q.pop();
int v=top.v;
if(vis[v]) continue;
vis[v]=1;
for(int i=0;i<G[v].size();++i)
{
edge e=G[v][i];
if(dis[e.v]>dis[v]+e.dist)
{
dis[e.v]=dis[v]+e.dist;
Q.push(edge(e.v,dis[e.v]));
}
}
}
ll ans=0;
for(int i=1;i<=n;i++)
ans+=dis[i];
return ans;
}
void addedge(int u,int v,ll w)
{
G[u].push_back(edge(v,w));
}
int main()
{
int T,P,Q;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&P,&Q);
for(int i=1;i<=Q;i++)
{
scanf("%d%d%d",&u[i],&v[i],&d[i]);
addedge(u[i],v[i],d[i]);
}
ll ans=dijkstra(P);
for(int i=1;i<=P;i++) G[i].clear();
for(int i=1;i<=Q;i++) addedge(v[i],u[i],d[i]);
ans+=dijkstra(P);
for(int i=1;i<=P;i++) G[i].clear();
printf("%lld\n",ans);
}
return 0;
}
我的代码,TLE(表示不明白为什么会TLE):
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#define inf 1000000010
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
ll dis[maxn];
bool vis[maxn];
struct edge
{
int v;
ll dist;
bool operator < (const edge& a) const
{
return dist>a.dist;
}
};
vector<edge> G1[maxn],G2[maxn];
void dijkstra(int n)
{
priority_queue<edge> Q;
for(int i=1;i<=n;i++)
{
dis[i]=inf;
vis[i]=0;
}
dis[1]=0;
edge s;
s.dist=0; s.v=1;
Q.push(s);
while(!Q.empty())
{
edge top=Q.top(); Q.pop();
int v=top.v;
if(vis[v]) continue;
vis[v]=1;
for(int i=0;i<G1[v].size();++i)
{
edge e=G1[v][i];
if(dis[e.v]>dis[v]+e.dist)
{
dis[e.v]=dis[v]+e.dist;
edge k;
k.v=e.v; k.dist=dis[e.v];
Q.push(k);
}
}
}
ll ans=0;
for(int i=1;i<=n;i++)
{
ans+=dis[i];
dis[i]=inf;
vis[i]=0;
}
dis[1]=0;
Q.push(s);
while(!Q.empty())
{
edge top=Q.top(); Q.pop();
int v=top.v;
if(vis[v]) continue;
vis[v]=1;
for(int i=0;i<G2[v].size();++i)
{
edge e=G2[v][i];
if(dis[e.v]>dis[v]+e.dist)
{
dis[e.v]=dis[v]+e.dist;
edge k;
k.v=e.v; k.dist=dis[e.v];
Q.push(k);
}
}
}
for(int i=1;i<=n;i++)
ans+=dis[i];
printf("%lld\n",ans);
}
int main()
{
int T,P,Q;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&P,&Q);
for(int i=1;i<=Q;i++)
{
int u,v,d;
scanf("%d%d%d",&u,&v,&d);
edge e;
e.dist=d; e.v=v;
G1[u].push_back(e);
e.v=u;
G2[v].push_back(e);
}
dijkstra(P);
for(int i=1;i<=P;i++)
{
G1[i].clear();
G2[i].clear();
}
}
return 0;
}
点击阅读全文
更多推荐
所有评论(0)