leetcode-go/oj/360-2019-spring/24point.js

/* eslint-disable semi */
function solve24point(...numbers) {
    class Operation {
        constructor(val, rank, exp) {
            this.val = val;
            this.rank = rank;
            this.exp = exp;
        }

        plus(that) {
            return new Operation(this.val + that.val, 0, `${this.exp}+${that.exp}`);
        }

        minus(that) {
            let exp;
            if (that.rank === 0) exp = `${this.exp}-(${that.exp})`;
            else exp = `${this.exp}-${that.exp}`;
            return new Operation(this.val - that.val, 0, exp);
        }

        mult(that) {
            let exp;
            if (this.rank === 0 && that.rank === 0) exp = `(${this.exp})*(${that.exp})`;
            else if (this.rank === 0) exp = `(${this.exp})*${that.exp}`;
            else if (that.rank === 0) exp = `${this.exp}*(${that.exp})`;
            else exp = `${this.exp}*${that.exp}`;
            return new Operation(this.val * that.val, 1, exp);
        }

        div(that) {
            let exp;
            if (this.rank === 0 && that.rank !== 2) exp = `(${this.exp})/(${that.exp})`;
            else if (this.rank === 0) exp = `(${this.exp})/${that.exp}`;
            else if (that.rank !== 2) exp = `${this.exp}/(${that.exp})`;
            else exp = `${this.exp}/${that.exp}`;
            return new Operation(this.val / that.val, 1, exp);
        }
    }

    const eps = 0.000001;

    function dfs(nums) {
        if (nums.length === 1 && Math.abs(nums[0].val - 24.0) < eps) return nums[0].exp;
        for (let i = 0; i < nums.length - 1; i++) {
            for (let j = i + 1; j < nums.length; j++) {
                let next = new Array(nums.length - 2);
                for (let k = 0, idx = 0; k < nums.length; k++) {
                    if (k == i || k == j) continue;
                    next[idx++] = nums[k];
                }
                let p = nums[i], q = nums[j];
                let tmp = [p.plus(q), p.minus(q), q.minus(p), p.mult(q)];
                if (q.val > 0) tmp.push(p.div(q));
                if (p.val > 0) tmp.push(q.div(p));
                for (let i = 0; i < tmp.length; i++) {
                    next.push(tmp[i]);
                    const res = dfs(next);
                    if (res !== null) return res;
                    next.pop();
                }
            }
        }
        return null;
    }

    const nums = new Array(numbers.length);
    for (let i = 0; i < numbers.length; i++) {
        nums[i] = new Operation(numbers[i], 2, numbers[i]);
    }
    return dfs(nums);
}


// eslint-disable-next-line no-console
const log = console.log.bind(console);

log(solve24point(2, 4, 1, 8));
log(solve24point(2, 4, 6, 8));
log(solve24point(1, 2, 5, 8));
log(solve24point(1, 7, 8, 8));
log(solve24point(6, 6, 6, 6, 6));
log(solve24point(1, 5, 5, 5));
log(solve24point(2, 5, 5, 10));
log(solve24point(1, 4, 5, 6));
log(solve24point(6, 9, 9, 10));