PAT1018——最短路加DFS

http://pat.zju.edu.cn/contests/pat-a-practise/1018

在杭州各个点，有很多自助自行车的点，最大容纳点为CMAX,但比较适合的情况是CMAX/2，

现在从起点到终点，在最短路的情况下，调整各个车站的自行车数量使之CMAX/2，问开始时要提供的自行车最少与最后剩余的自行车最少的路径

最短路+DFS

在DFS时加上一个剪枝 if(lenth>dis[first])return ;就可以通过所有点

#include<stdio.h>
#define MAX 0x3ffffff
int cmax,N,end,M;
int node[599];
int map[599][599];
int path[599];

int rpath[599];
int in,out,nowstep,rstep;

bool hash[599][599];
int shortlen;

int dis[505];
bool use[505];
int ok=0;


void dijk( )
{
    int n=N,m;
    int min,rj,i,j;
    for(j=0;j<=n;j++)
    {
        dis[j]=map[0][j];
        use[j]=0;
    }
    use[0]=1;
    for(i=0;i<n;i++)
    {
        min=MAX;
        for(j=0;j<=n;j++)
        {
            if(use[j]==1)continue;
            if(dis[j]<min)
            {
                min=dis[j];
                rj=j;
            }
        }
        
        use[rj]=1;
        for(j=0;j<=n;j++)
        {
            if(use[j]==1)continue;
            if(dis[j]>dis[rj]+map[rj][j])
                dis[j]=dis[rj]+map[rj][j];
        }
    }

    shortlen=dis[end];
}


void print(){
    int i;
    printf("%d",in);
    printf(" 0");
    for(i=1;i<=rstep;i++){
        printf("->%d",rpath[i]);
    }
    printf(" %d\n",out);
}

void cal(){
    int tin=99999999,tout=999999999;
    int i,add=0;
    for(i=1;i<=nowstep;i++){
        add+=node[path[i]]-cmax/2;
        if(add<tin)tin=add;
    }
    if(tin>0)tin=0;
    else tin=-tin;
    tout=add+tin;

    if((tin<in)||((tin==in)&&tout<out)){
        in=tin;
        out=tout;
        if(in==0&&tout==0)ok=1;
        rstep=nowstep;
        for(i=1;i<=nowstep;i++){
            rpath[i]=path[i];
        }
    }

    //test
    /*
    printf("%d",tin);
    printf(" 0");
    for(i=1;i<=nowstep;i++){
        printf("->%d",path[i]);
    }
    printf(" %d\n",tout);
    */
}
void dfs(int step,int first,int lenth)
{
    if(ok==1||lenth>dis[first])return ; //lenth>dis[first] 是个关键的剪枝，不加的话最后一个点会超时
    if(lenth==shortlen&&first==end){
        nowstep=step;
        cal();
    }

    int j;
    for(j=1;j<=N;j++){
        if(hash[first][j]==1||map[j][first]==MAX)continue;
        hash[first][j]=hash[j][first]=1;
        path[step+1]=j;
        dfs(step+1,j,lenth+map[first][j]);
        hash[first][j]=hash[j][first]=0;
    }
}

int main()
{
    while(scanf("%d%d%d%d",&cmax,&N,&end,&M)!=EOF){
        int i,j,temp;
        ok=0;
        in=99999999;
        out=99999999;

        for(i=0;i<=N;i++){
            for(j=0;j<=N;j++){
                map[i][j]=MAX;
                hash[i][j]=0;
            }
        }

        for(i=1;i<=N;i++){
            scanf("%d",&node[i]);
        }
        

        int ll,rr,v;
        for(i=1;i<=M;i++){
            scanf("%d%d%d",&ll,&rr,&v);
            map[ll][rr]=map[rr][ll]=v;
        }
        dijk();
        dfs(0,0,0);
        print();
    }

    return 0;
}