二进制枚举二叉树子集
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | #include <iostream> #include <cstring> #include <vector> using namespace std; const int MAX = 6; struct Tree{ double L, R; //树的左边和右边的长度 Tree():L(0), R(0){} }; double width, w[MAX], sum[1<<MAX]; int vis[1<<MAX]; vector<Tree> tree[1<<MAX]; //枚举出来的结点可以组合成多个树 void dfs( int subset); //枚举二叉树 int main(){ freopen( "input.txt" , "r" , stdin); int T; cin >> T; while (T--){ int s; scanf( "%lf%d" , &width, &s); for ( int i=0; i<s; i++){ scanf( "%lf" , &w[i]); } //初始化 memset(sum, 0, sizeof (sum)); for ( int i=1; i<(1<<s); i++){ tree[i].clear(); //清空树 int k = 1; for ( int j=0; j<s; j++){ //算出当前集合所有结点的重量和 if (k & i){ sum[i] += w[j]; } k = (k << 1); } } //进行枚举 int root = (1<<s) - 1; memset(vis, 0, sizeof (vis)); dfs(root); //得到最大的答案 double ans = -1; for ( int i=0; i<tree[root].size(); i++){ ans = max(tree[root][i].L + tree[root][i].R, ans); } printf( "%.10lf\n" , ans); } } void dfs( int subset){ if (vis[subset]) return ; vis[subset] = 1; bool hasChild = false ; for ( int left = ((subset-1) & subset); left > 0; left = ((left-1) & subset)){ //枚举集合 hasChild = true ; int right = subset ^ left; //left 的补集 double l = sum[right] / sum[subset]; //左臂长度 double r = 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 + l, tree[right][j].L - r); t.R = max(tree[right][j].R + r, tree[left][i].R - l); if (t.L + t.R < width){ //如果满足条件,枚举结果加一 tree[subset].push_back(t); } } } } if (!hasChild){ //叶子结点的左右臂为零 tree[subset].push_back(Tree()); } } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步