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
			}