POJ 1485

题意:n个加工厂,m个原料厂,原料厂地址与加工厂一致,m<=n,求最佳分配,使得每个加工厂到达与它最近的那个原料厂,且使得总距离最小。

题解:dp[i][j]为前i个厂,j个原料厂最佳分配的总代价。另外,对于加工厂区间[a,b]用原料厂去覆盖的方法应该是建立在最中间那个点上,这样左右工厂数相等,到达它的距离之和最小。

dp[i][j]=min(dp[k][j-1]+ds[k+1][i]),厂址位置就在(k+1+i)/2位置,可以用pre[i][j]记录该原料厂相关的区间的左端点,最后再算一下就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=202,K=202,inf=0x3f3f3f3f;
int d[N],dp[N][K],pre[N][K],ds[N][N];
struct data
{
    int depot,from,to;
    data(){}
    data(int _d,int _f,int _t){depot=_d;from=_f;to=_t;}
    bool operator<(const data &ne)const
    {
        return depot<ne.depot;
    }
}po[N];
int cal(int i,int j)
{
    int ans=0,t=d[(i+j)>>1];
    while(i<=j)
        ans+=abs(t-d[i++]);
    return ans;
}
int main()
{
    int n,m,ca=0;
    while(scanf("%d%d",&n,&m),n||m)
    {
        for(int i=1; i<=n; i++)
            scanf("%d",d+i);
        memset(dp,0x3f,sizeof(dp));
        memset(pre,-1,sizeof(pre));
        memset(ds,0,sizeof(ds));
        d[n+1]=inf;d[0]=-inf;
        for(int i=1;i<n;i++)
            for(int j=i+1;j<=n;j++)
                ds[i][j]=cal(i,j);
        for(int i=1;i<=n;i++)
            dp[i][1]=ds[1][i];
        for(int i=2; i<=n; i++)
            for(int j=2; j<=m; j++)
                for(int k=1,tp;k<i;k++)
                    if(dp[i][j]>(tp=dp[k][j-1]+ds[k+1][i]))
                        dp[i][j]=tp,pre[i][j]=k;
        printf("Chain %d\n",++ca);
        int nu=0;
        for(int i=n,j=m,k;j!=0;)
        {
            k=pre[i][j];
            if(k==-1)
                k=0;
            po[nu++]=data((k+i+1)/2,k+1,i);
            i=k;
            j--;
        }
        sort(po,po+nu);
        for(int i=0; i<nu; i++)
        {
            printf("Depot %d at restaurant %d serves restaurant",i+1,po[i].depot);
            if(po[i].from==po[i].to) printf(" %d\n",po[i].from);
            else printf("s %d to %d\n",po[i].from,po[i].to);
        }
        printf("Total distance sum = %d\n\n",dp[n][m]);
    }
    return 0;
}


你可能感兴趣的:(POJ 1485)