UVa 1354 - Mobile Computing（二叉树 + DFS）

来源：互联网发布：淘宝主图用什么软件做编辑：程序博客网时间：2024/06/05 02:37

题目链接

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4100

题目大意

给定多个二叉树的叶子，每个节点有一个重量，两个节点可以用一个木棍连接，一个木棍的一头可以连接一个叶子节点，或者另一根木棍。

要保证每个木棍平衡，符合胡克定律。

解题过程

刚开始写的，状态是用一个 vector 储存的，每次删去两个节点，继续归并其他的。
然后来回操作非常麻烦，也一直 WA。

于是百度了下别人的博客，他是每一层 dfs 都开一个 vector 数组储存状态，然后以引用传递，传给下级。

每层一个 vector 储存当前的状态，避免了来回操作一个数组造成 bug。

题目分析

主要是注意下，这里 dfs 时每层储存状态。

AC代码

#include<bits/stdc++.h>using namespace std;const int MAX = 112;const double EPS = 1e-8;struct node{    double first;    double second;    double right;    node (double first, double second, double right):first(first),second(second),right(right){}};int n;double width, ans = -1;vector<node> status;node connect(node a, node b){    double w = a.second + b.second;    double la = b.second / w, lb = a.second / w;    node ans = node(max(a.first + la, -lb + b.first), w,max(b.right + lb, -la + a.right));    return ans;}void solve(vector<node> &status){    if (status.size() == 1)    {        double t = status[0].first + status[0].right;        if (t <= width + EPS)            ans = max(ans, t);        return;    }    for (int i = 0; i < status.size(); i++)    {        for (int j = 0; j < status.size(); j++)        {            if (i == j)                continue;            vector<node> nstatus;            for (int k = 0; k < status.size(); k++)            {                if (k == i || k == j)                    continue;                nstatus.push_back(status[k]);            }            node a = status[i], b = status[j];            node x = connect(a, b);            if (x.first + x.right > width + EPS)                continue;            nstatus.push_back(x);            solve(nstatus);        }    }}int main(){    int T;    scanf("%d", &T);    while (T--)    {        status.clear();        ans = -1;        scanf("%lf", &width);        scanf("%d", &n);        while (n--)        {            double w;            scanf("%lf", &w);            status.push_back(node(0.0, w, 0.0));        }        solve(status);        if (ans < 0)            puts("-1");        else            printf("%.16lf\n", ans);    }}

0 0