牛客小白月赛 22 D. 收集纸片

链接

https://ac.nowcoder.com/acm/contest/4462/D

题意

求从起点 s s s 出发经过 n n n 个点后会到起点的最短距离

思路

状压DP

d p [ i ] [ j ] dp[i][j] dp[i][j] 为( i i i 的二进制数某位为 1 1 1 则代表经过该点)并最终到达点 j j j 的最短距离

转移方程: d p [ i ] [ j ] = m i n ( d p [ i ] [ j ] , d p [ i − ( 1 < < j ) ] [ k ] + d i s t [ j ] [ k ] ) dp[i][j]=min(dp[i][j],dp[i-(1<<j)][k]+dist[j][k]) dp[i][j]=min(dp[i][j],dp[i(1<<j)][k]+dist[j][k])

因为最后还要回到起点,所以

r e s = m i n ( d p [ ( 1 < < ( n + 1 ) ) − 1 ] [ i ] + d i s t [ s ] [ i ] ) res=min(dp[(1<<(n+1))-1][i]+dist[s][i]) res=min(dp[(1<<(n+1))1][i]+dist[s][i])

代码

#include<bits/stdc++.h>
using namespace std;
int n;
int dp[2100][11];
int dist[11][11];
struct node {
    int x,y;
}pap[11];
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;cin>>T;
    while(T--) {
        int row,col;
        cin>>row>>col;
        cin>>pap[0].x>>pap[0].y;
        cin>>n;
        for(int i=1;i<=n;i++) cin>>pap[i].x>>pap[i].y;
        for(int i=0;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                dist[j][i]=dist[i][j]=abs(pap[j].x-pap[i].x)+abs(pap[j].y-pap[i].y);
        int lim=(1<<(n+1))-1;
        memset(dp,0x3f,sizeof dp);
        dp[1][0]=0;
        for(int i=1;i<=lim;i++)
            for(int j=0;j<=n;j++)
                if(i&(1<<j))
                    for(int k=0;k<=n;k++)
                        if((i-(1<<j))&(1<<k))
                            dp[i][j]=min(dp[i][j],dp[i-(1<<j)][k]+dist[k][j]);
        int res=0x3f3f3f3f;
        for(int i=1;i<=n;i++) res=min(res,dp[lim][i]+dist[0][i]);
        cout<<"The shortest path has length "<<res<<endl;
    }
    return 0;
}
©️2020 CSDN 皮肤主题: 创作都市 设计师:CSDN官方博客 返回首页