【CCF 交通规划 立flag中】

这两天划水了……我要跪……

第一种思路，先以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;
}