1018 Public Bike Management

#include <iostream>
#include <string.h>
#include <algorithm>
#include <cstdio>
#include <math.h>
#include <string>
#include <map>
#include <string>
#include <stack>
#include <queue>
#include <set>
#include <vector>
using namespace std;
int cmax,n,ps,m;
const int inf=0x3fffffff;
struct NODE{
	int weight;
}node[1010];
struct EDAGE{
	int v;
	int w;
}edage[101000000];
vector<EDAGE> G[1010];
int d[1010];
int flag[1010]={0};
vector<int> pre[10100];
void findmin(int& u,int &minx)
{
	for(int j=0;j<=n;j++)
	{
		if(d[j]<minx&&flag[j]==0)
		{
			u=j;
			minx=d[j];
		}
	}
}
void Dij()
{
	fill(d,d+1010,inf);
	d[0]=0;
	for(int i=0;i<=n;i++)
	{
		int u=-1,minx=inf;
		findmin(u,minx);
		if(u==-1) return ;
		flag[u]=1;
		for(int j=0;j<G[u].size();j++)
		{
			int v=G[u][j].v;
			if(d[v]>d[u]+G[u][j].w)
			{
				d[v]=d[u]+G[u][j].w;
				pre[v].clear();
				pre[v].push_back(u);
			}
			else if(d[v]==d[u]+G[u][j].w)
			{
				pre[v].push_back(u);
			}
		}
	}
}
int good=inf;
vector<int> path,temp;
int minbring=0;
void DFS(int x)
{
	if(x==0)
	{
		temp.push_back(x);
		int take=0,bring=0;//
		for(int i=temp.size()-2;i>=0;i--)
		{
			int need=cmax/2-node[temp[i]].weight;//need>0 要补 need<0 带走
			if(bring-need<0)//不够
			{
				take+=need-bring;
				bring=0;
			}
			else//够/超过
			{
				bring=bring-need;
			}
		}
		if(take<0)
			take=0;
		if(take<good||(take==good&&bring<minbring))
		{
			good=take;
			path=temp;
			minbring=bring;
		}
		temp.pop_back();
		return ;
	}
	temp.push_back(x);
	for(int i=0;i<pre[x].size();i++)
	{
		DFS(pre[x][i]);
	}
	temp.pop_back();
}
int main()
{
	cin >> cmax >> n >> ps >> m;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&node[i].weight);
	}
	for(int i=1;i<=m;i++)
	{
		int num1,num2,num;
		cin >> num1 >> num2 >> num;
		edage[i].v=num2;
		edage[i].w=num;
		G[num1].push_back(edage[i]);
        edage[i].v=num1;
        G[num2].push_back(edage[i]);
	}
	Dij();
	DFS(ps);
	if(good>0)
	{
		printf("%d ",good);
		for(int i=path.size()-1;i>=0;i--)
		{
			printf("%d",path[i]);
			if(i!=0)
			{
				printf("->");
			}
		}
		printf(" ");
		printf("%d\n",minbring);
	}
	else
	{
		printf("%d ",good);
		for(int i=path.size()-1;i>=0;i--)
		{
			printf("%d",path[i]);
			if(i!=0)
			{
				printf("->");
			}
		}
		printf(" ");
		printf("%d",minbring);
	}
}

一开始只过了三个测试点
原因是以为带回和带走之前肯定有一个是0，这个是对题目理解不到位
改了之后过了前四个测试点
中间两个测试点是
只更新了带走的数量，没有更新带回数量
没有判断带走数量相同但带回数量更小时也要更新路径
还有最后两个测试点
路是双向的