Uva1354 天平难题

在刘汝佳老师的代码上加了些注释

// UVa1354 Mobile Computing
// Rujia Liu
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;

struct Tree {   //因为我们只关心长度啊~  所以只记录长度就可以啦~【关心什么就记录什么！】
  double L, R; // distance from the root to the leftmost/rightmost point
  Tree():L(0),R(0) {}
};

const int maxn = 6;

int n, vis[1<<maxn];      //二进制枚举的vis数组也用二进制标记
double r, w[maxn], sum[1<<maxn];
vector<Tree> tree[1<<maxn];

void dfs(int subset) {    //一定要写成这种二进制的形式 如果转换成十进制，不同的二进制(子集)可能对应相同的十进制！
  if(vis[subset]) return;
  vis[subset] = true;

  bool have_children = false;
  for(int left = (subset-1)&subset; left; left = (left-1)&subset) {   //原来遍历一个集合的子集是这么写的啊...
    have_children = true;

    int right = subset^left;
    double d1 = sum[right] / sum[subset];   //left(right)的距离反而是right(left)的重量除以总重量
    double d2 = sum[left] / sum[subset];
    //这一层的算出来以后算下一层的
    dfs(left); dfs(right);

    for(int i = 0; i < tree[left].size(); i++)      //左子集也有很多种组织的方法 
      for(int j = 0; j < tree[right].size(); j++) { //右子集也有很多种组织的方法！
        Tree t;
        t.L = max(tree[left][i].L + d1, tree[right][j].L - d2);   //可能左(右)天平的最右(左)端反而比右(左)天平的最左（右）端还长!
        t.R = max(tree[right][j].R + d2, tree[left][i].R - d1);
        if(t.L + t.R < r) tree[subset].push_back(t);      //似乎从下到上会越来越多，但也没办法了不是
      }
  }

  if(!have_children) tree[subset].push_back(Tree());  //空的话就代表着 就挂着这个吊坠（实际上此时只能对应原子集只有一个挂坠的情况了）
}

int main() {
  int T;
  scanf("%d", &T);
  while(T--) {
    scanf("%lf%d", &r, &n);   //r是房间宽度    n是挂坠的个数
    for(int i = 0; i < n; i++) scanf("%lf", &w[i]);
    for(int i = 0; i < (1<<n); i++) {
      sum[i] = 0;
      tree[i].clear();
      for(int j = 0; j < n; j++)
        if(i & (1<<j)) sum[i] += w[j];
    }

    int root = (1<<n)-1;    //二进制枚举是吧
    memset(vis, 0, sizeof(vis));
    dfs(root);

    double ans = -1;
    for(int i = 0; i < tree[root].size(); i++)
      ans = max(ans, tree[root][i].L + tree[root][i].R);
    printf("%.10lf\n", ans);
  }
  return 0;
}