最小重量机器设计---回溯

问题描述

设某一机器由n个部件组成，每一种部件都可以从m个不同的供应商处购得。设 wij 是从供应商j 处购得的部件i的重量，cij 是相应的价格。试设计一个回溯算法，给出总价格不超过d的最小重量机器设计。对于给定的机器部件重量和机器部件价格，计算总价格不超过cost的最小重量机器设计。

算法思想

通过使用回溯来选择合适的机器使得在总价格不超过d时得到的机器重量最小。
1、设置中间变量 tempw,tempc,x[],保存搜索过程中的重量,价格,商家编号
2、minw,best[],保存最优结果
3、step枚举零件,大于n时停止搜索，因为只有n个零件
4、搜索过程需要回溯，因为试探性搜索寻找最优解

#include<stdio.h>
#define N 1000
#define INF 0x3f3f3f3f
int w[N][N],c[N][N];
int best[N],x[N];
int cost,n,m;
int tempw,tempc,minw=INF;

void dfs(int step)///step为第几个部件
{
    if(step>n)
    {
        minw=tempw;
        for(int i=1; i<=n; i++)
            best[i]=x[i];
        return;
    }
    for(int i=1; i<=m; i++)///枚举商家
    {
        if(tempw+w[step][i]<minw && tempc+c[step][i]<=cost)
        {
            x[step]=i;
            tempw+=w[step][i];
            tempc+=c[step][i];
            dfs(step+1);
            tempw-=w[step][i];
            tempc-=c[step][i];
        }
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&cost);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            scanf("%d",&w[i][j]);
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            scanf("%d",&c[i][j]);
    dfs(1);///从第一个部件开始找
    printf("%d\n",minw);///最小重量
    for(int i=1; i<=n; i++)///最优解的商家
        printf("%d ",best[i]);
    return 0;
}