UVA - 1354 Mobile Computing dfs

来源：互联网发布：cnn tensorflow 编辑：程序博客网时间：2024/05/24 06:38

题目大意：有n个砝码，要求放在天平上，每个天平的长度是1，要求放上砝码后，使得天平的水平长度达到最大

解题思路：因为砝码的数量只有6个，所以可以用一个二进制数来表示状态
枚举每种状态下的结果，左端的长度用负数表示，右端的长度用正数表示
枚举该状态时，先把当下的子状态都枚举了，然后取其最大值或者最小值即可。。。好难解释，具体看代码

#include<cstdio>#include<cmath>#include<algorithm>#include<vector>using namespace std;#define maxn (1 << 6)struct Node{    double l, r;    Node() {}    Node(double ll, double rr) {        l = ll, r = rr;    }};vector<Node> node[maxn];double w[maxn], sum[maxn], len;int n, vis[maxn];int bitcount(int x) {    return x ? x % 2 + bitcount(x/2): 0;}void dfs(int s) {    if(vis[s])        return ;    vis[s] = 1;    if(bitcount(s) == 1) {        node[s].push_back(Node(0,0));        return ;    }    for(int l = (s - 1) & s; l > 0; l = (l - 1) & s) {        int r = s ^ l;        double Sum = sum[l] + sum[r];        dfs(l), dfs(r);        for(int i = 0; i < node[l].size(); i++)            for(int j = 0; j < node[r].size(); j++) {                double ll = min(-sum[r]/Sum + node[l][i].l, sum[l] / Sum + node[r][j].l);                double rr = max(-sum[r]/Sum + node[l][i].r, sum[l] / Sum + node[r][j].r);                node[s].push_back(Node(ll,rr));            }    }}void solve() {    double ans = -1;    int s = (1 << n) - 1;    dfs(s);    for(int i = 0; i < node[s].size(); i++)         if(node[s][i].r - node[s][i].l < len)            ans = max(ans, node[s][i].r - node[s][i].l);    if(ans == -1)        printf("-1\n");    else        printf("%.16lf\n", ans);}void init() {        scanf("%lf%d", &len, &n);        for(int i = 0; i < n; i++)            scanf("%lf", &w[i]);        for(int i = 0; i < (1 << n); i++) {            sum[i] = 0;            vis[i] = 0;            node[i].clear();            for(int j = 0; j < n; j++)                if(i & (1 << j))                    sum[i] += w[j];        }}int main() {    int test;    scanf("%d", &test);    while(test--) {        init();        solve();    }    return 0;}

0 0