poj1511小灶第三次作业(dijstra)

版权声明:转我原创记得说你是我的脑残粉哟 https://blog.csdn.net/zjy2015302395/article/details/50976480

历时三天,终于ac。。。第一次这么晚睡觉呵呵。。。

大哭

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=109960#problem/F


题意:

直接翻译:有个出发点,然后有很多个公交站台,从一个站台到另一个站台需要花钱,求从初始点到所有站台,然后再回来 最少需要多少钱
浓缩成题意:有n个点,m条边的有向图 ,求从初始点到所有点的长度加上从所有点回到起始点的长度


耐心看完前面的吐槽,后面有对ac代码的耐心讲解,嘿嘿。。。吐舌头

队友说是模板题,10分钟就搞定了。。。然而我竟然没注意到数据量是1000000,开二维数组mle

代码如下:

#include <cstring>
#include <cstdio>
#include <queue>
#include <iostream>
#include <vector>
#include <deque>
#include <utility>
#include <functional>
const int maxnum = 1000;
#define maxn 10000

using namespace std;
int dis[maxn][maxn];
int book[maxnum] ,n;
typedef pair <int ,int> pii;
priority_queue <pii,vector<pii> ,greater <pii> > pq;

void ppqq(int lon1[])
{
    int node ,flag = 0 ,lon ;
    while (!pq.empty())
        {
            lon = pq.top().first;
            node = pq.top().second;

            pq.pop();

            if(book[node] == 1)
                continue;
            lon1[node] = lon;
            book[node] = 1;
            flag++;

            if(flag == n)
                break;
            for (int i = 1 ; i <= n ;i++)
            {
                if( !book[i] && dis[node][i])
                    pq.push(make_pair(lon+dis[node][i] , i));
            }
        }
    return ;
}

void ppq(int lon2[])
{
    int node ,flag = 0 ,lon ;
    while (!pq.empty())
        {
            lon = pq.top().first;
            node = pq.top().second;

            pq.pop();

            if(book[node] == 1)
                continue;

            lon2[node] = lon;
            book[node] = 1;
            flag++;

            if(flag == n)
                break;
            for (int i = 1 ; i <= n ;i++)
            {
                if( !book[i] && dis[i][node])
                    pq.push(make_pair(lon+dis[i][node] , i));
            }
        }

    return ;
}


int main()
{
    int T;
    int lon1[maxnum],lon2[maxnum] ;
    scanf("%d",&T);

    while (T--)
    {
        int m,u,v,w;
        scanf("%d%d",&n,&m);
        memset(dis,0,sizeof(dis));
        memset(lon1,0,sizeof(lon1));
        memset(lon2,0,sizeof(lon2));
        memset(book,0,sizeof(book));

        for (int i = 0 ; i < m ;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            dis[u][v] = w;
        }


        pq.push(make_pair(0,1));
        ppqq(lon1);

        while (!pq.empty())
            pq.pop();
        memset(book,0,sizeof(book));


        pq.push(make_pair(0,1));
        ppq(lon2);

        int ans = 0;
        for (int i = 1 ; i<= n ;i++)
            {
                ans += lon1[i] + lon2[i];
               printf("%d   %d\n",lon1[i],lon2[i]);
            }

        printf("%d\n",ans);
    }
    return 0;
}
也对啊,这么大数据量开这么多数组,1000000 * 1000000,不mle我都不信。然后就用了结构体。。。然后tle了。。

也是醉了啊,写的时候也是各种错误,查了好久,尤其两次调用,已经改变了的结构体的成员值不知道怎么办了,后来又开了个数组减回去,虽然tle也算白写吧

代码如下:


#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
const int maxnum = 1000010;

using namespace std;

struct node
{
    int u,v,w;
    bool operator < (const node &a) const
    {
        return w > a.w;
    }

};

priority_queue <node> pq;
int ans = 0, n,m,book[maxnum];
struct node edg[maxnum];
int s[maxnum];

void dj1(struct node edg[])
{
    int node ,flag = 0 ,lon ;
    while (!pq.empty())
        {
            lon = pq.top().w;
            node = pq.top().v;

            pq.pop();

            if(book[node] == 1)
                continue;
            ans += lon;
          //  printf("dj1 %d\n",lon);
            book[node] = 1;
            flag++;

            if(flag == n)
                break;
            for (int i = 1 ; i <= m ;i++)
            {
               // printf("qishidian%d   %d\n",edg[i].u,edg[i].v);
                if( !book[edg[i].v] && edg[i].u == node)
                    {
                       // printf("qishidian%d   %d\n",edg[i].u,edg[i].v);
                        edg[i].w += lon ;
                        s[i] += lon;
                        pq.push(edg[i]);
                    }
            }
        }
    return ;
}

void dj2(struct node edg[])
{
    int node ,flag = 0 ,lon ;
    while (!pq.empty())
        {
            lon = pq.top().w;
            node = pq.top().u;

            pq.pop();

            if(book[node] == 1)
                continue;

            ans += lon;
            //printf("dj2 nod = %d  lon =   %d\n",node ,lon);
            book[node] = 1;
            flag++;

            if(flag == n)
                break;
            for (int i = 1 ; i <= m ;i++)
            {
                 //printf("一定要找到你%d   %d    %d\n",edg[i].u,edg[i].v,edg[i].w);
                if( !book[edg[i].u]  && edg[i].v == node)
                   {
                   //    printf("目前的%d   %d    %d\n",edg[i].u,edg[i].v,edg[i].w);
                       edg[i].w += lon;
                     //  printf("改变后%d   %d    %d\n",edg[i].u,edg[i].v,edg[i].w);
                        pq.push(edg[i]);
                   }
            }
        }

    return ;
}


int main()
{
    int t;
    scanf("%d",&t);

    while (t--)
    {
        ans = 0;

        memset(s,0,sizeof(s));
        memset(book,0,sizeof(book));
        scanf("%d%d",&n,&m);

        for (int i = 1 ; i <= m ;i++)
            scanf("%d%d%d",&edg[i].u,&edg[i].v,&edg[i].w);
        edg[0].u = 1;
        edg[0].v = 1;
        edg[0].w = 0;

        pq.push(edg[0]);
        dj1(edg);

        while (!pq.empty())
            pq.pop();
        memset(book,0,sizeof(book));
        for (int i = 1 ; i <= m ;i++)
            edg[i].w -= s[i];

        pq.push(edg[0]);
        dj2(edg);

        printf("%d\n",ans);
    }
}
哦哦哦、、、你还在看,那就给你看ac的代码吧,说实话,这回真是模板了,就像刚说的,结构体成员值改变了就是不可逆的了,下次调用不好玩,所以可以结合一下啊,push的都是pair《int,int》,这样子就不会改变了,然后第二次调用同一个函数的话可以少写好多的说,怎么做到呢,只能将这个图图整个反过来,这时候求出的1到所有点的最短路就是要求的原图的所有点到起始点的最短距离了。虽然是挑战上的模板 ,但还是说一下算法,以后也方便想来。。。vector是个不定长的数组,里面每个成员都是个结构体,结构体只存终点和距离,那起点呢 ? 你会发现vector又给了个数组。。。目的就是把起点都是 i 的点都放在vector【 i 】里 ,比如点3 到2 5 8点都有距离,vector【3】【0】,vector【3】【1】 , vector【3】【2】就分别保存着这几个结构体了,这样知道源点(3)就可以快速不多循环的找到这几个和3相连的点点。。。。pair这三个代码都是一样的,first值是保存源点到这个点的距离,second是这是哪个点,然后,优先队列,小顶堆,肯定能做到最短路了 ,d数组就是记录到每个点的最短路长,初始化INT_MAX,如果更新到这个点,发现曾经更新的比这次的小,那这次pop后就可以直接玩下一个队头元素了,保证d里面放的是最短路径长度,然后,,,,没什么了吧。。。。

给代码:

#include <climits>
#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#include <functional>
#include <utility>
#define maxnum 1000010
using namespace std;

struct edg
{
    int to;
    int dis;
};

typedef pair <int ,int > pii;
vector <edg> from[maxnum];
int d[maxnum];
int u[maxnum],v[maxnum],w[maxnum];
priority_queue < pii , vector <pii> ,greater <pii> > pq;

void dij1()
{
    while ( !pq.empty())
    {

        int ss = pq.top().first;
        int node = pq.top().second;
        pq.pop();

        if(d[node] < ss )
            continue;
        for (int i = 0 ; i < from[node].size();i++)
        {
            edg e = from[node][i];
            if(d[e.to] > d[node] + e.dis)
            {
                d[e.to] = d[node] + e.dis;
                pq.push(make_pair(d[e.to],e.to));
            }
        }

    }
}

int main()
{
    int t;
    scanf("%d",&t);

    while (t--)
    {
        int n , m ;
        scanf("%d%d",&n,&m);
        long long int ans = 0;
        for (int i = 0 ; i < m ;i++)
         scanf("%d%d%d",&u[i],&v[i],&w[i]);

        for (int i = 0 ; i < m ;i++)
        {
            edg e;
            e.to=v[i];
            e.dis=w[i];
            from[u[i]].push_back(e);
        }
        for (int i = 1 ; i <= n ;i++)
            d[i] = INT_MAX;

        pq.push(make_pair(0,1));
        d[1] = 0;
        dij1();
        for (int i = 1 ;i <= n ;i++)
            ans += d[i];

        for (int i = 0 ;i < m ;i++)
            from[i].clear();
       for (int i = 1 ; i <= n ;i++)
            d[i] = INT_MAX;
        for (int i = 0 ; i < m ;i++)
        {
            edg e;
            e.to=u[i];
            e.dis=w[i];
            from[v[i]].push_back(e);
        }
        pq.push(make_pair(0,1));
        d[1] = 0;
        dij1();
        for (int i = 1 ;i <= n ;i++)
            ans += d[i];

        printf("%I64d\n",ans);
         for (int i = 0 ;i < m ;i++)
            from[i].clear();
    }
    return 0;
}
要清空vector啊!!!!wa了一发,大半夜的多吓人啊。。。

阅读更多
想对作者说点什么?
相关热词

博主推荐

换一批

没有更多推荐了,返回首页