Hatena::Grouptopcoder

hama_DU@TopCoderへの道

2012-02-16SRM398 (Practice)

SRM 398 CountExpressions

|  SRM 398 CountExpressions - hama_DU@TopCoderへの道 を含むブックマーク はてなブックマーク -  SRM 398 CountExpressions - hama_DU@TopCoderへの道

http://www.topcoder.com/stat?c=problem_statement&pm=8157


コーディングフェーズ

  • (サンプルを斜め読みして)げ、めんどくさい系だ
    • 掛け算は先に計算しなきゃいけないし・・・(←間違い)
  • 数字とオペランドの配列を与えると計算結果を返す関数をつくろう
    • 掛け算はどう処理しよう?
      • スタックに積んでいって、掛け算が来たらスタックから一個取り出して掛け算して積もう
    • で、最後にスタックの残りを全部足す。と
    • マイナスが来たらどうする?
      • * -1 してスタックに積もう。-aの足し算をしたことと同じになる
  • 数字の列は4つだから、next_permutationしよう
    • ライブラリを貼り付ける
    • 同じ数列が生成されるから、List<Integer>をマップに突っ込んで既に処理したかどうか調べよう
  • できた。
    • サンプル1が合わない・・・
    • 5 - 3 * 5 - 3 ←このケースが生成されていない。でもこれって 7 じゃなくて -13 じゃん。
      • 何か見落としがあるのか・・・?(ここではじめて問題を斜め読みする)
  • 前から先に計算すればいいだけだった・・・
    • 泣く泣くロジックを修正。提出

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.Stack;

public class CountExpressions {
	public int calc(int[] l, int[] operand) {
		int len = l.length;
		int ans = l[0];
		for (int i = 0 ; i < 3 ; i++) {
			if (operand[i] == 2) {
				ans *= l[i+1];
			} else if (operand[i] == 1){
				ans -= l[i+1];
			} else {
				ans += l[i+1];
			}
		}
		return ans;
	}
	
	public int calcExpressions(int x, int y, int val) {
		int cnt = 0;
		int[] li = new int[]{x, x, y, y};
		Arrays.sort(li);
		
		Set<List<Integer>> processed = new HashSet<List<Integer>>();
		
		do {
			List<Integer> nl = new ArrayList<Integer>();
			for (int i = 0 ; i < 4 ; i++) {
				nl.add(li[i]);
			}
			if (processed.contains(nl)) {
				continue;
			}
			processed.add(nl);
			
			for (int ptn = 0 ; ptn < 27 ; ptn++) {
				int[] op = new int[3];
				int p = ptn;
				for (int i = 0 ; i < 3 ; i++) {
					op[i] = p % 3;
					p /= 3;
				}
				if (calc(li, op) == val) {
					cnt++;
				}
			}
		} while (next_permutation(li));
		
		return cnt;
	}

	
	public boolean next_permutation(int[] num) {
		int len = num.length;
		int x = len - 2;
		while (x >= 0 && num[x] >= num[x+1]) {
			x--;
		}
		if (x == -1) return false;

		int y = len - 1;
		while (y > x && num[y] <= num[x]) {
			y--;
		}
		int tmp = num[x];
		num[x] = num[y];
		num[y] = tmp;
		java.util.Arrays.sort(num, x+1, len);
		return true;
	}
}