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[];
int map[][];
int path[];

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

bool hash[][];
int shortlen;

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

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

        use[rj]=;
        for(j=;j<=n;j++)
        {
            if(use[j]==)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("");
    for(i=;i<=rstep;i++){
        printf("->%d",rpath[i]);
    }
    printf(" %d\n",out);
}

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

    if((tin<in)||((tin==in)&&tout<out)){
        in=tin;
        out=tout;
        if(in==&&tout==)ok=;
        rstep=nowstep;
        for(i=;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==||lenth>dis[first])return ; //lenth>dis[first] 是个关键的剪枝，不加的话最后一个点会超时
    if(lenth==shortlen&&first==end){
        nowstep=step;
        cal();
    }

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

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

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

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

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

    return ;
}