UVA 662 Fast Food

UVA_662

    首先我們考慮區間[i,j]內如果建一個餐館應該建在哪個位置，比較容易證明應該是建在中間的位置，如果是偶數個元素，中間兩個位置隨便一個都可以。這樣我們就可以預先處理出區間[i,j]內建一個餐館的最小總路程A[i][j]了。

    我們可以用f[i][j]表示第i個餐館服務到第j個位置所需的最小總路程，那麼f[i][j]=min{f[i-1][k]+A[k+1][j]}，其中i-1<=k<j。

#include<stdio.h>
#include<string.h>
#define MAXD 210
#define MAXK 40
#define INF 0x3f3f3f3f
int N, K, f[MAXK][MAXD], p[MAXK][MAXD], a[MAXD], b[MAXD];
int A[MAXD][MAXD], s[MAXD];
int init()
{
int i, j, k, temp;
    scanf("%d%d", &N, &K);
if(!N && !K)
return 0;
    s[0] = 0;
for(i = 1; i <= N; i ++)
    {
        scanf("%d", &a[i]);
        s[i] = a[i] + s[i - 1];
    }
for(i = 1; i <= N; i ++)
for(j = i; j <= N; j ++)
        {
            k = (i + j) / 2;
            temp = s[j] - s[k] - (j - k) * a[k];
            temp += (k - i) * a[k] - s[k - 1] + s[i - 1];
            A[i][j] = temp;
        }
return 1;
}
void printpath(int k, int n)
{
int t = p[k][n];
if(k != 1)
        printpath(k - 1, t);
    printf("Depot %d at restaurant %d serves restaurants %d to %d\n", k, (t + 1 + n) / 2, t + 1, n);
}
void solve()
{
int i, j, k, min;
    memset(f, 0x3f, sizeof(f));
for(k = 1; k <= N; k ++)
        f[1][k] = A[1][k];
for(i = 2; i <= K; i ++)
for(j = i; j <= N; j ++)
for(k = i - 1; k < j; k ++)
            {
if(f[i - 1][k] + A[k + 1][j] < f[i][j])
                {
                    f[i][j] = f[i - 1][k] + A[k + 1][j];
                    p[i][j] = k;
                }
            }
    printpath(K, N);
    printf("Total distance sum = %d\n", f[K][N]);
}
int main()
{
int t = 0;
while(init())
    {
        printf("Chain %d\n", ++ t);
        solve();
        printf("\n");
    }
return 0;
}