最短最优升级路径(某公司面试编程题)

来源:互联网 发布:起点西游记知乎 编辑:程序博客网 时间:2024/06/06 07:23

最短最优升级路径
时间限制:C/C++语言 1000MS;其他语言 3000MS
内存限制:C/C++语言 65536KB;其他语言 589824KB
题目描述:
游戏网站提供若干升级补丁,每个补丁大小不一,玩家要升级到最新版,如何选择下载哪些补丁下载量最小。
输入
第一行输入 第一个数为用户版本 第二个数为最新版本,空格分开
接着输入N行补丁数据 第一个数补丁开始版本 第二个数为补丁结束版本 第三个数为补丁大小,空格分开
输出
对于每个测试实例,输出一个升级路径以及最后实际升级的大小

样例输入
1000 1050
1000 1020 50
1000 1030 70
1020 1030 15
1020 1040 30
1030 1050 40
1040 1050 20
样例输出
1000->1020->1040->1050(100)

思路:
晚上帮某学长做了一道某公司编程题,考的单源最短路问题,数据都很离散,我第一次用的map水过,感觉很神奇。然后思考了一下少一层的floyd其实就是可以求单源最短路问题。

#include<bits/stdc++.h>using namespace std;map<pair<int,int>,int>M;map<int,int>dir;vector<int>ans;set<int>se;const int inf=0x7ffffff;int s,e;void dfs(int x,int sum){    if(x==s) return;    for(set<int>::iterator i=se.begin();i!=se.end();i++){      if(sum-M[make_pair(*i,x)]==M[make_pair(s,*i)]){           dfs(*i,sum-M[make_pair(*i,x)]);           break;       }    }    ans.push_back(x);}int main(){     scanf("%d%d",&s,&e);     int a,b,c;     M.clear();     se.clear();     dir.clear();     ans.clear();     while(~scanf("%d%d%d",&a,&b,&c)){        M[make_pair(a,b)]=c;        M[make_pair(b,a)]=c;        se.insert(a);se.insert(b);     }     for(set<int>::iterator i=se.begin();i!=se.end();i++){       for(set<int>::iterator j=se.begin();j!=se.end();j++){         if(M[make_pair(s,*i)]==0)  M[make_pair(s,*i)]=inf;         if(M[make_pair(s,*j)]==0)  M[make_pair(s,*j)]=inf;         if(M[make_pair(*j,*i)]==0)  M[make_pair(*j,*i)]=inf;         if(M[make_pair(s,*i)]>M[make_pair(s,*j)]+M[make_pair(*j,*i)]){            M[make_pair(s,*i)]=M[make_pair(s,*j)]+M[make_pair(*j,*i)];         }       }     }     dfs(e,M[make_pair(s,e)]);     printf("%d",s);     for(int i=0;i<ans.size();i++) printf("->%d",ans[i]);     printf("(%d)\n",M[make_pair(s,e)]);}
1 0
原创粉丝点击