Hatena::Grouptopcoder

hama_DU@TopCoderへの道

2011-03-13SRM494、SRM498など(DIV1)

SRM498 第二問(450pt) FoxStones

|  SRM498 第二問(450pt) FoxStones - hama_DU@TopCoderへの道 を含むブックマーク はてなブックマーク -  SRM498 第二問(450pt) FoxStones - hama_DU@TopCoderへの道

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

各マスを、ポイントからの距離の組み合わせによってグループ分けするだけ。


import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

public class FoxStones {

	public int[] Dx = {1, 0, -1, 0};
	public int[] Dy = {0, 1, 0, -1};

	public long MODULO = 1000000009;

	public int getCount(int N, int M, int[] sx, int[] sy) {
		int length = sx.length;
		ArrayList<StringBuffer> arr = new ArrayList<StringBuffer>();

		for (int x = 0 ; x < N ; x++) {
			for (int y = 0 ; y < M ; y++) {
				arr.add(new StringBuffer());
			}
		}

		for (int i = 0 ; i < length ; i++) {
			int cx = sx[i] - 1;
			int cy = sy[i] - 1;
			for (int x = 1 ; x <= 200 ; x++) {
				int nx = cx - x;
				int ny = cy - x;
				for (int d = 0 ; d < 4 ; d++) {
					for (int dd = 0 ; dd < x * 2 ; dd++) {
						nx += Dx[d];
						ny += Dy[d];
						if (nx >= 0 && ny >= 0 && nx < N && ny < M) {
							arr.get(ny * N + nx).append(i).append("_").append(x).append(",");
						}
					}
				}
			}
		}


		Map<String, Long> ctmap = new HashMap<String, Long>(40000, 1.0f);
		for (int x = 0 ; x < N ; x++) {
			for (int y = 0 ; y < M ; y++) {
				String s = arr.get(y * N + x).toString();
				if (s.length() > 0) {
					if (ctmap.containsKey(s)) {
						ctmap.put(s, ctmap.get(s) + 1L);
					} else {
						ctmap.put(s, 1L);
					}
				}
			}
		}

		long result = 1L;
		System.out.println(ctmap.keySet().size());
		for (String key : ctmap.keySet()) {
			long l = ctmap.get(key);
			for (int n = 2 ; n <= l ; n++) {
				result = (result * n) % MODULO;
			}
		}
		return (int)result;
	}
}

簡単と思ったがそんなことはなかった。

上のコードだと時間的にはギリギリ間に合うものの、メモリを使いきってしまう。

もうちょっと効率のいい解き方があるのかも。