题目链接
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);
}
}