2021-04-06 PAT A1087

测试点2不过的原因,是更新路径条数的时候,不是加上它前驱的所有路径条数,而是只加一
纯DIJSTRA实现
这道题做的真的好幸福啊,因为是都没怎么调,就做出来了,让我对DIJSTRA的认识又进一步了

#include<cstdio>
#include<utility>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn = 26 * 26 * 26 + 5, INF = 0x3fffffff;
vector<pair<int, int> > graph[maxn];//first是id second是cost
int happyValue[maxn], level[maxn], dis[maxn], city_num, happy_sum[maxn], pre[maxn], path[maxn], star_id;//level用来记录层数,最后层数最少的就是平均幸福值最低的
bool vis[maxn];
int getID(char* name) {
    int id = 0;
    for (int i = 0; i < 3; i++)
        id = id * 26 + name[i] - 'A';
    return id;
}
void print_name(int id) {
    char name[6]; name[3] = '\0';
    for (int i = 2; i >= 0; i--) {
        int temp = id % 26;
        id /= 26;
        name[i] = temp + 'A';
    }
    printf("%s", name);
}
void DIJSTRA(int source) {
    fill(dis, dis + maxn, INF);
    priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > q;
    q.push(pair<int, int>(0, source)); dis[source] = 0; path[source] = 1; level[source] = 0;
    pre[source] = source;
    for (int i = 0; i < city_num; i++) {
        if (q.empty())break;
        pair<int, int> temp = q.top(); q.pop();
        int u = temp.second; vis[u] = true;
        //print_name(u); printf(" ");
        for (int j = 0; j < graph[u].size(); j++) {
            int v = graph[u][j].first, cost = graph[u][j].second;
            //print_name(v); printf("\n");
            if (!vis[v]) {
                if (dis[v] > dis[u] + cost) {
                    level[v] = level[u] + 1;
                    dis[v] = dis[u] + cost;
                    happy_sum[v] = happy_sum[u] + happyValue[v];
                    pre[v] = u;
                    q.push(pair<int, int>(dis[v], v));
                    path[v] = path[u];
                }
                else if (dis[v] == dis[u] + cost) {
                    path[v] += path[u];
                    if (happy_sum[v] < happy_sum[u] + happyValue[v]) {
                        level[v] = level[u] + 1;
                        happy_sum[v] = happy_sum[u] + happyValue[v];
                        pre[v] = u;
                    }
                    else if (happy_sum[v] == happy_sum[u] + happyValue[v] && level[u] + 1 < level[v]) {//层数越少,那么快乐平均值越高
                        level[v] = level[u] + 1;
                        pre[v] = u;
                    }
                }
            }
        }
    }
}
void DFS(int source) {
    if (source == star_id) print_name(source);
    else {
        DFS(pre[source]);
        printf("->");
        print_name(source);
    }

}
int main() {
    char city1[6], city2[6], strdes[6];//最后这个用来记录出发的地点
    int k, cost; scanf("%d %d %s", &city_num, &k, strdes); getchar();
    star_id = getID(strdes);
    for (int i = 1; i < city_num; i++) {
        scanf("%s", city1);
        scanf("%d", &happyValue[getID(city1)]);
    }
    for (int i = 0; i < k; i++) {
        scanf("%s %s %d", city1, city2, &cost); getchar();
        int id1 = getID(city1), id2 = getID(city2);
        graph[id1].push_back(pair<int, int>(id2, cost));
        graph[id2].push_back(pair<int, int>(id1, cost));
    }
    DIJSTRA(star_id);
    char end_name[] = "ROM";
    int end = getID(end_name);
    printf("%d %d %d %d\n", path[end], dis[end], happy_sum[end], happy_sum[end] / level[end]);
    DFS(end);printf("\n");
    
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,099评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,828评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,540评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,848评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,971评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,132评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,193评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,934评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,376评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,687评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,846评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,537评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,175评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,887评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,134评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,674评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,741评论 2 351

推荐阅读更多精彩内容