UVA 10317 Equating Equations （状态压缩）

题意：

给一个等式，但是这个等式不一定是正确的，要你对等式中的数字重新排序，使得等式成立。等式只有+和-，数字个数小于16。

分析：

由于数字只有16，总共有2^16种可能，所以可以用状态压缩求解。
以a + b - c = d - e为例子。
1. 我们把等式右边的各项都换到左边，a + b - c - d + e = 0
2. 把+项和-项放一起，就变成(a + b + e) - (c + d) = 0，所以所有数字的和sum应该为偶数。如果不是剪枝。
3. 然后(a + b + e) = (c + d)
4. 令+为1，-为0，枚举所有的状态。
5. 然后我们只要找到多少个-项，状态压缩枚举同样多的数，且和正好是sum / 2，那等式就能成立了。

AC代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <string>
#include <sstream>
#include <cstdlib>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
vector<int> rec; //保存数字
vector<char> op; //保存操作符
int pos , neg; //记录正负数的个数
void init() {
    rec.clear();
    op.clear();
}
void preProc(string str) {
    string buf;
    stringstream ss(str);
    int x;
    bool equal = false;
    pos = 1, neg = 0;
    while(ss >> buf) {
        if(isdigit(buf[0])) {
            x = atoi(buf.c_str());
            rec.push_back(x);
        }else {
            op.push_back(buf[0]);
            if(buf[0] == '=')
                equal = true, neg++;
            else if(equal ^ (buf[0] == '+'))
                pos++;
            else
                neg++;
        }
    }
}
int getSum() {
    int sum = 0;
    for(int i = 0; i < rec.size(); i++) {
        sum += rec[i];
    }
    return sum;
}
void print_state(int st, int cnt) {
    vector<int> postive, negative; //新建两个向量用于保存正数和负数

    for(int i = 0; i < rec.size(); i++) {
        if(st & (1 << i)) //保存正数
            postive.push_back(rec[i]);
        else //保存负数
            negative.push_back(rec[i]);
    }

    int np = 0, nn = 0; //保存正数的个数，保存负数的个数
    bool equal = false;
    if(cnt == pos) {
        printf("%d", postive[np++]);
        for(int i = 0; i < op.size(); i++) {
            if(op[i] == '=') {
                equal = true; printf(" = %d", negative[nn++]);
            }else if(op[i] == '+') {
                equal ? printf(" + %d",negative[nn++]) : printf(" + %d",postive[np++]);
            }else {
                equal ? printf(" - %d",postive[np++]) : printf(" - %d",negative[nn++]);
            }
        }
    }else {
        printf("%d", negative[nn++]);
        for(int i = 0; i < op.size(); i++) {
            if(op[i] == '=') {
                equal = true; printf(" = %d", postive[np++]);
            }else if(op[i] == '+') {
                equal ? printf(" + %d",postive[np++]) : printf(" + %d",negative[nn++]);
            }else {
                equal ? printf(" - %d",negative[nn++]) : printf(" - %d",postive[np++]);
            }
        }
    }
    puts("");
}
void solve() {
    int posSum, negSum, cnt;
    int sum = getSum();
    if(sum & 1) { //如果sum为奇数不符合条件
        puts("no solution");
        return;
    }

    sum = sum / 2;

    int end = 1 << rec.size();
    for(int i = 0; i < end; i++) { //遍历最多 2^16种情况
        posSum = negSum = cnt = 0;
        for(int j = 0; j < rec.size(); j++) {
            if(posSum > sum || neg > sum) break;
            if(i & (1 << j))
                posSum += rec[j] , cnt++;
            else
                negSum += rec[j];
        }
        if(posSum == negSum && (cnt == pos || cnt == neg)) {
            print_state(i, cnt);
            return ;
        }
    }
    puts("no solution");
}
int main() {
    string str;
    while(getline(cin, str)) {
        init();
        preProc(str);
        solve();
    }
    return 0;
}