hdu 4284 floyed+狀態壓縮dp


開始看這道題其實就是一個旅行商問題具體求法都差不多:抽出關鍵點,找到最短距離,然後狀態壓縮,這道題要注意的是1.起點可能沒有包含在必須去的點裏  2  在必須去的點中如果能打工  則必須打工賺錢(感覺題意不是這樣的 不過只能這樣了) 注意初始化dp是可能打工也可能不打工。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<map>;
using namespace std;

#define INF 0x3f3f3f3f

int dp[1<<16][16];
int Map[110][110];
struct Node
{
    int ii;
    int pay;
    int get;    
}point[20];
int swap(int &a,int &b)
{
    int t;
    t=a;
    a=b;
    b=t;
    return 0;
}
int min(int a,int b)
{
    return a<b?a:b;
}
int main()
{
    int n,m,i,j,T,money;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&money);
        memset(Map,INF,sizeof(Map));
        for(i=1;i<=m;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            Map[a][b]=Map[b][a]=min(Map[a][b],c);
        }
        for(int k=1;k<=n;k++)
        for(i=1;i<=n;i++)
         for(j=1;j<=n;j++)
         if(Map[i][j]>Map[i][k]+Map[k][j])
        Map[i][j]=Map[i][k]+Map[k][j];
        
        int k;
        scanf("%d",&k);
        for(i=0;i<k;i++)
        {
            scanf("%d%d%d",&point[i].ii,&point[i].get,&point[i].pay);
            if(point[i].ii==1)
            {
                swap(point[i].ii,point[0].ii);
                swap(point[i].pay,point[0].pay);
                swap(point[i].get,point[0].get);
            }
        }
        if(point[0].ii!=1)
        {
            point[k].ii=1;
            point[k].pay=0;
            point[k].get=0;
            swap(point[k].ii,point[0].ii);
            swap(point[k].pay,point[0].pay);
            swap(point[k].get,point[0].get);
            k++;
        }
        memset(dp,-1,sizeof(dp));
        if(money>=point[0].pay) dp[1][0]=money-point[0].pay+point[0].get;
        else dp[1][0]=money;
        int t=(1<<k);
        for(i=1;i<t;i++)
        {
            for(j=0;j<k;j++)
            {
                if((i&(1<<j))==0) continue;
                if(dp[i][j]<0) continue;
                for(int x=0;x<k;x++)
                {
                    if(x==j) continue;
                    if((i&(1<<x))!=0) continue;
                    if(dp[i][j]>=Map[point[j].ii][point[x].ii]+point[x].pay)
                    {
                        dp[i|(1<<x)][x]=max(dp[i|(1<<x)][x],dp[i][j]+point[x].get-Map[point[j].ii][point[x].ii]-point[x].pay);
                    }
                }
            }
        }
        int flash=0;
        for(i=0;i<k;i++)
        {
            if(dp[t-1][i]>=Map[point[i].ii][1]) flash=1;
        }
        if(flash) printf("YES\n");
        else printf("NO\n");
    }
    return 0;    
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章