不错的网试题目(集合dp)

floyd + 路径dp + 集合dp :

//for (int s = i & (i - 1); s; s = (s - 1) & i)  枚举i的每一个非空真子集s,若包含i需修改s初始化

#ifndef SOLVER_H_
#define SOLVER_H_

#include <bits/stdc++.h>
using namespace std;

const int kMaxN = 55;
const int kMaxM = 10;
const int kMaxS = 1024;

struct DeliveryCost {
  int u;
  int v;
  int cost;
};

class Solver {
 public:
  double solve(int n,
               vector<DeliveryCost> cost_e,
               vector<int> employees,
               vector<int> cost_b) {
    /*********begin*********/
    static int floyd[kMaxN][kMaxN];
    memset(floyd, 0x3f, sizeof(floyd));
    for (int i = 1; i <= n; ++i) {
      floyd[i][i] = 0;
    }
    int m = cost_e.size();
    for (int i = 0; i < m; ++i) {//所有员工进行floyd最短路
      int a = cost_e[i].u;
      int b = cost_e[i].v;
      floyd[a][b] = cost_e[i].cost;
      floyd[b][a] = cost_e[i].cost;
    }
    for (int k = 1; k <= n; ++k) {
      for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
          floyd[i][j] = min(floyd[i][j], floyd[i][k] + floyd[k][j]);
        }
      }
    }

    static int f[kMaxS][kMaxN];//f[i][j]:选了i集合,当前在j点的最小花费
    memset(f, 0x3f, sizeof(f));
    int num_employees = employees.size();
    int num_states = 1 << num_employees;
    //f初始化
    for (int i = 1; i <= n; ++i) f[0][i] = 0;
    for (int i = 0; i < num_employees; ++i) f[1 << i][employees[i]] = 0;
    
    for (int i = 0; i < num_states; ++i) {
      for (int j = 1; j <= n; ++j) {
        for (int s = i & (i - 1); s; s = (s - 1) & i) {//对i的子集进行dp转移
          f[i][j] = min(f[i][j], f[s][j] + f[i - s][j]);
        }
      }
      for (int j = 1; j <= n; ++j) {//对j的路径进行dp转移
        for (int k = 1; k <= n; ++k) {
          f[i][j] = min(f[i][j], f[i][k] + floyd[k][j]);
        }
      }
    }

    static int g[kMaxS];    //g[i]:选了i集合,且i集合中有一个特殊员工与boss连接的最小花费
    memset(g, 0x3f, sizeof(g));
    for (int i = 0; i < num_states; ++i) {
      for (int j = 0; j < num_employees; ++j) {
        if ((i >> j) & 1) {
          g[i] = min(g[i], f[i][employees[j]] + cost_b[j]);
        }
      }
    }

    static int h[kMaxS];    //h[i]:选了i集合,且i集合中至少一个特殊员工与boss连接的最小花费
    for (int i = 0; i < num_states; ++i) {
      h[i] = g[i];
      for (int s = i & (i - 1); s; s = (s - 1) & i) {//枚举i的子集进行dp转移
        h[i] = min(h[i], h[s] + g[i - s]);
      }
    }

    return h[num_states - 1];
    /*********end*********/
  }
};

#endif

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值