这两天划水了……我要跪……
第一种思路,先以1为原点做dijkstra找最短路,然后用类似找最小生成树的方式生成最后的值,如果连接一个点的时候花费不是最短的,那么就不连接,反正最后总能连上的……交上去就wa了一发,因为这样做有个问题…比如测试用例为
3 3
1 2 2
2 3 1
1 3 1
的时候,算出来的结果就不对了……先挂一发代码
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
#define maxn 1010
#define maxm 0x3f3f3f3f
int n,m;
int a[maxn][maxn], d[maxn],tmp[maxn], vis[maxn];
int ans;
int min(int x,int y){return x<y?x:y;}
void dijkstra()
{
d[1]=0;
memset(tmp,0,sizeof(tmp));
int num=n;
while(num--)
{
int j=maxm,k=-1;
for(int i=2;i<=n;i++)
{
if(!tmp[i]&&d[i]<j)
{
j=d[i];
k=i;
}
}
if(k<0) k=1,j=0;
tmp[k]=1;
for(int i=1;i<=n;i++)
{
if(i!=k&&a[k][i]<maxm&&d[k]+a[k][i]<d[i])
{
d[i]=d[k]+a[k][i];
}
}
}
// for(int i=1;i<=n;i++) cout<<d[i]<<" ";cout<<endl;
}
void search()
{
vis[1]=1;
int x=1;
while(1)
{
for(int i=1;i<=n;i++)
{
if(d[x]+a[x][i]==d[i])
{
if(!vis[i])
{
ans+=a[x][i];
vis[i]=1;
}
// cout<<i<<" "<<a[x][i]<<endl;
}
}
int ts=maxm,k=-1;
for(int i=1;i<=n;i++)
{
if(!vis[i]) continue;
for(int j=1;j<=n;j++)
{
if(j==i) continue;
// cout<<j<<" "<<a[i][j]<<" "<<d[j]<<endl;
if(a[i][j]<ts && a[i][j]+d[i]==d[j])
{
ts=a[i][j];
k=j;
}
}
}
x=k;
if(x==-1) break;
ans+=ts;
if(vis[x]) ans-=ts;
// cout<<k<<" "<<ts<<endl;
vis[x]=1;
}
}
void init()
{
for(int i=0;i<maxn;i++) for(int j=0;j<maxn;j++) a[i][j]=maxm;
for(int i=0;i<maxn;i++) d[i]=maxm;
memset(vis,0,sizeof(vis));
}
int main()
{
while(cin>>n>>m)
{
int x,y,z;
init();
for(int i=0;i<m;i++)
{
cin>>x>>y>>z;
a[x][y]=min(min(a[y][x],a[x][y]), z);
a[y][x]=a[x][y];
}
dijkstra();
ans=0;
search();
cout<<ans<<endl;
}
return 0;
}
这里还没有考虑节点数较大的情况。当节点数大于1000的数量级的时候,就要考虑用前向星来记录地图状态,或者用map?
最近犯相思病,有点疲懒……
看了其他人的思路…刚开始用vector做…居然编译不过…试了好多次才相信用vector太大了……黑人问号脸,于是乖乖用前向星…但是到了80分依然是超时?于是还要优化diskstra…然而很懒……
#include <cstdio>
#include <iostream>
#include <string>
#include <vector>
#include <cstring>
using namespace std;
#define maxn 10010
#define maxm 0x3f3f3f3f
int n,m;
int top;
int d[maxn],tmp[maxn],ans[maxn];
struct node
{
int u,v,c,next;
}b[100010];
int a[maxn];
int min(int x,int y){return x<y?x:y;}
void dijkstra()
{
d[1]=0;
for(int i=0;i<maxn;i++) tmp[i]=0;
int num=n;
while(num--)
{
int j=maxm,k=-1;
for(int i=2;i<=n;i++)
{
if(!tmp[i]&&d[i]<j)
{
j=d[i];
k=i;
}
}
if(k<0) k=1,j=0;
tmp[k]=1;
for(int i=a[k];i>=0;i=b[i].next)
{
int y=b[i].v, z = b[i].c;
if((y!=k)&&(d[k]+z<=d[y]))
{
d[y]=d[k]+z;
ans[y]=min(ans[y],z);
}
}
}
}
void init()
{
for(int i=0;i<maxn;i++) d[i]=ans[i]=maxm, a[i]=-1;
}
void add(int x,int y, int z)
{
b[top].u=x,b[top].v=y,b[top].c=z,b[top].next=a[x];
a[x]=top++;
}
int main()
{
scanf("%d%d",&n,&m);
top=0;
int x,y,z;
init();
for(int i=0;i<m;i++)
{
cin>>x>>y>>z;
add(x,y,z);
add(y,x,z);
}
dijkstra();
int ret=0;
for(int i=2;i<=n;i++) ret+=ans[i];
printf("%d\n",ret);
return 0;
}