1018. Public Bike Management (30)

#include<iostream>#include<vector>#include<algorithm>using namespace std;const int INF = 0x3fffffff;const int MAX = 1000;vector<int> pre[MAX];//存放最短路径上节点的前驱，可能不止一条路径 int G[MAX][MAX];bool visited[MAX]={false};int weight[MAX],need[MAX];//节点的现有自行车数量，及需要的自行车数量负数代表需要，反之多余 int d[MAX];int weightmax,n,index,m;//求need数组void calneed(){for(int i=1;i<n;i++){need[i]=weight[i]-weightmax/2;}} void Dijkstra(int s){fill(d,d+MAX,INF);d[s]=0;for(int i=0;i<n;i++){int u=-1,min=INF;//找到最小的d[u]for(int j=0;j<n;j++){if(visited[j]==false&&d[j]<min){u=j;min=d[j];}}if(u==-1) return;//节点不连通visited[u]=true;for(int v=0;v<n;v++){if(visited[v]==false&&G[u][v]!=INF){if(d[u]+G[u][v]<d[v]){   d[v]=d[u]+G[u][v];   pre[v].clear();   pre[v].push_back(u);    }    else if(d[u]+G[u][v]==d[v]){    pre[v].push_back(u);    }}}}} //先选择需要派遣的自行车最少路径，在此基础上如果有多条路径，//再选择回收最少的路径 int sentmin=INF,backmin=INF;//派遣最少，回收最少 vector<int> optpath, temppath;//最优路径、临时路径//从目标节点递归访问直到起点（PBMC）void DFS(int v){//递归边界if(v==0){temppath.push_back(v);int sent=0,back=0,temp=0;//存放路径temppath的sent，back值    vector<int> sum;//利用need数组计算路径temppath上从起点（pbmc）到各节点的need数组和sum.clear();        //找到最小值（显然如果是负的，最小值的绝对值就是从PBMC出发是应该带的//自行车数量（不然路上不够用，尽管后面可能有的节点剩的多，但是回不了头）//如果sum数组值都非负，那么很好开始的时候不用带自行车，sent=0；//这种情况下找back最小的路径 for(int i=temppath.size()-2;i>=0;i--){//temppath存放路径是倒序的,最后一个必为零，没必要计算,同时省去为need[0]及weight[0]赋值 temp+=need[temppath[i]];sum.push_back(temp);//sum存放路径前k节点的need数组之和  }if(*min_element(sum.begin(),sum.end())<0){sent=-*min_element(sum.begin(),sum.end());}else{sent=0;//路径顺序上任意缺车节点前面多余的车足够补充缺少的车 }back=temp+sent;//sent意思就是PCMB派出的车数  if(sent<sentmin){//选派遣车辆最少的 sentmin=sent;backmin=back;optpath=temppath;}else if(sent==sentmin&&back<backmin){//派遣车辆都为最少，选取回收最少的 backmin=back;optpath=temppath;}temppath.pop_back();} //递归式temppath.push_back(v);for(int i=0;i<pre[v].size();i++){DFS(pre[v][i]);//递归v的前驱节点 } temppath.pop_back(); }int main(){int n_temp;scanf("%d%d%d%d",&weightmax,&n_temp,&index,&m);n=n_temp+1;for(int i=1;i<n;i++){scanf("%d",&weight[i]);}calneed();int itemp,jtemp,weighttemp;fill(G[0],G[0]+MAX*MAX,INF);for(int i=0;i<m;i++){scanf("%d%d%d",&itemp,&jtemp,&weighttemp);G[itemp][jtemp]=G[jtemp][itemp]=weighttemp;} Dijkstra(0);/*for(int i=1;i<n;i++){for(int j=0;j<pre[i].size();j++){printf("%d ",pre[i][j]);}printf("\n");}*/DFS(index);printf("%d 0",sentmin);for(int i=optpath.size()-2;i>=0;i--){printf("->%d",optpath[i]);}printf(" %d\n",backmin);return 0;} 

1、开始的时候pre[max]数组找到了，但是DFS一运行就崩溃

由此发现一个vector的使用问题

错误使用方法：运行崩溃

#include<iostream>#include<vector>#include<algorithm>using namespace std;vector<int> a,b;int main(){a[0]=1;a[1]=2;a[2]=3;cout<<a[2];}

正确使用方法：

#include<iostream>#include<vector>#include<algorithm>using namespace std;vector<int> a,b;int main(){a.push_back(1);a[1]=2;a[2]=3;cout<<a[2];}

2、backmin在这里忘了更新

if(sent<sentmin){//选派遣车辆最少的 sentmin=sent;backmin=back;optpath=temppath;}

然后又在外面用了一个标记tag=0更新backmin，乍看没问题，不过当tag=1，之后，再遇到sent<sentmin变化的back等不到更新，结果后四个检测点过不了，8分没了

back=temp+sent;//sent意思就是PCMB派出的车数 

if(tag==0) backmin=back; if(sent<sentmin){//选派遣车辆最少的 sentmin=sent;//backmin=back;optpath=temppath;}else if(sent==sentmin&&back<backmin){//派遣车辆都为最少，选取回收最少的 backmin=back;optpath=temppath;

                        tag==1}