fork(3) download
copy 
  1. /* paiza POH! vol.2
  2.  * result:
  3.  * http://p...content-available-to-author-only...a.jp/poh/paizen/result/b86d786a8604d0245966f6caf3666192
  4.  * author: Leonardone @ NEETSDKASU
  5.  */
  6. import java.util.*;
  7. import java.lang.*;
  8. import java.io.*;
  9.  
  10. class Main
  11. {
  12. 	public static void main (String[] args) throws java.lang.Exception
  13. 	{
  14. 		Paiza.getInstance().resolve(new MyResolver());
  15. 	}
  16. }
  17.  
  18. class MyResolver extends Paiza.Resolver
  19. {
  20. 	private int[][] space2right = null;
  21. 	private int[][] space2bottom = null;
  22. 	private int[][] result = null;
  23. 	private int[] maxSpace2right = null;
  24. 	private int[] maxSpace2bottom = null;
  25. 	private int[] space2rightLen = null;
  26. 	private int[] space2bottomLen = null;
  27. 	private int maxSpace2rightLen = 0;
  28. 	private int maxSpace2bottomLen = 0;
  29.  
  30. 	@Override
  31. 	public void setHome(Paiza.Home home) {
  32. 		super.setHome(home);
  33.  
  34. 		int allSpaceCount = 0;
  35.  
  36. 		space2right = new int[home.getH()][home.getW()];
  37. 		maxSpace2right = new int[home.getH()];
  38. 		space2rightLen = new int[301];
  39. 		maxSpace2rightLen = 0;
  40.  
  41. 		for (int y = 0; y < home.getH(); y++) {
  42. 			int count = 0, max = 0;
  43. 			for (int x = home.getW() - 1; x >= 0; x--) {
  44. 				if (home.isSpace(x, y)) {
  45. 					count++;
  46. 					allSpaceCount++;
  47. 					if (count > max) {
  48. 						max = count;
  49. 					}
  50. 				} else {
  51. 					if (count > 1) {
  52. 						space2rightLen[count]++;
  53. 					}
  54. 					count = 0;
  55. 				}
  56. 				space2right[y][x] = count;
  57. 			}
  58. 			if (count > 1) {
  59. 				space2rightLen[count]++;
  60. 			}
  61. 			maxSpace2right[y] = max;
  62. 			if (max > maxSpace2rightLen) {
  63. 				maxSpace2rightLen = max;
  64. 			}
  65. 		}
  66.  
  67. 		space2bottom = new int[home.getH()][home.getW()];
  68. 		maxSpace2bottom = new int[home.getW()];
  69. 		space2bottomLen = new int[301];
  70. 		maxSpace2bottomLen = 0;
  71.  
  72. 		for (int x = 0; x < home.getW(); x++) {
  73. 			int count = 0, max = 0;
  74. 			for (int y = home.getH() - 1; y >= 0; y--) {
  75. 				if (home.isSpace(x, y)) {
  76. 					count++;
  77. 					if (count > max) {
  78. 						max = count;
  79. 					}
  80. 				} else {
  81. 					if (count > 1) {
  82. 						space2bottomLen[count]++;
  83. 					}
  84. 					count = 0;
  85. 				}
  86. 				space2bottom[y][x] = count;
  87. 			}
  88. 			if (count > 1) {
  89. 				space2bottomLen[count]++;
  90. 			}
  91. 			maxSpace2bottom[x] = max;
  92. 			if (max > maxSpace2bottomLen) {
  93. 				maxSpace2bottomLen = max;
  94. 			}
  95. 		}
  96.  
  97. 		result = new int[301][301];
  98. 		result[1][1] = allSpaceCount + 1;
  99.  
  100. 	}
  101.  
  102. 	@Override
  103. 	public int resolve(Paiza.Widget widget) {
  104. 		if (result[widget.getS()][widget.getT()] > 0) {
  105. 			return result[widget.getS()][widget.getT()] - 1;
  106. 		}
  107. 		int count;
  108. 		if (widget.getS() > widget.getT()) {
  109. 			if (widget.getT() > 1) {
  110. 				count = resolve2bottom(widget);
  111. 			} else {
  112. 				count = resolveVBar(widget);
  113. 			}
  114. 		} else {
  115. 			if (widget.getS() > 1) {
  116. 				count = resolve2right(widget);
  117. 			} else {
  118. 				count = resolveHBar(widget);
  119. 			}
  120. 		}
  121. 		result[widget.getS()][widget.getT()] = count + 1;
  122. 		return count;
  123. 	}
  124.  
  125. 	private int resolveHBar(Paiza.Widget widget) {
  126. 		int count = 0;
  127. 		for (int i = widget.getT(); i <= maxSpace2rightLen; i++) {
  128. 			count += space2rightLen[i] * (i - widget.getT() + 1);
  129. 		}
  130. 		return count;
  131. 	}
  132.  
  133. 	private int resolveVBar(Paiza.Widget widget) {
  134. 		int count = 0;
  135. 		for (int i = widget.getS(); i <= maxSpace2bottomLen; i++) {
  136. 			count += space2bottomLen[i] * (i - widget.getS() + 1);
  137. 		}
  138. 		return count;
  139. 	}
  140.  
  141. 	private int resolve2right(Paiza.Widget widget) {
  142. 		int count = 0;
  143. 		for (int y = 0, ey = 0; y <= home.getH() - widget.getS(); y++) {
  144. 			for (int yy = y + widget.getS() - 1; yy > ey; yy--) {
  145. 				if (maxSpace2right[yy] < widget.getT()) {
  146. 					y = yy;
  147. 					ey = yy + 1;
  148. 					break;
  149. 				}
  150. 			}
  151. 			if (maxSpace2right[y] < widget.getT()) {
  152. 				continue;
  153. 			}
  154. 			ey = y + widget.getS() - 1;
  155. 			for (int x = 0; x <= home.getW() - widget.getT(); x++) {
  156. 				if (space2bottom[y][x] >= widget.getS()) {
  157. 					boolean placeable = true;
  158. 					for (int dy = 0; placeable && dy < widget.getS(); dy++) {
  159. 						if (space2right[y + dy][x] < widget.getT()) {
  160. 							placeable = false;
  161. 							x += space2right[y + dy][x];
  162. 						}
  163. 					}
  164. 					if (placeable) {
  165. 						count++;
  166. 					}
  167. 				} else if (space2right[y][x] < widget.getT()) {
  168. 					x += space2right[y][x];
  169. 				}
  170. 			}
  171. 		}
  172. 		return count;
  173. 	}
  174.  
  175. 	private int resolve2bottom(Paiza.Widget widget) {
  176. 		int count = 0;
  177. 		for (int x = 0, ex = 0; x <= home.getW() - widget.getT(); x++) {
  178. 			for (int xx = x + widget.getT() - 1; xx > ex; xx--) {
  179. 				if (maxSpace2bottom[xx] < widget.getS()) {
  180. 					x = xx;
  181. 					ex = xx + 1;
  182. 					break;
  183. 				}
  184. 			}
  185. 			if (maxSpace2bottom[x] < widget.getS()) {
  186. 				continue;
  187. 			}
  188. 			ex = x + widget.getT() - 1;
  189. 			for (int y = 0; y <= home.getH() - widget.getS(); y++) {
  190. 				if (space2right[y][x] >= widget.getT()) {
  191. 					boolean placeable = true;
  192. 					for (int dx = 0; placeable && dx < widget.getT(); dx++) {
  193. 						if (space2bottom[y][x + dx] < widget.getS()) {
  194. 							placeable = false;
  195. 							y += space2bottom[y][x + dx];
  196. 						}
  197. 					}
  198. 					if (placeable) {
  199. 						count++;
  200. 					}
  201. 				} else if (space2bottom[y][x] < widget.getS()) {
  202. 					y += space2bottom[y][x];
  203. 				}
  204. 			}
  205. 		}
  206. 		return count;
  207. 	}
  208.  
  209. }
  210.  
  211. final class Paiza
  212. {
  213. 	public static abstract class Resolver
  214. 	{
  215. 		protected Paiza.Home home;
  216.  
  217. 		public void setHome(Paiza.Home home) {
  218. 			this.home = home;
  219. 		}
  220.  
  221. 		public abstract int resolve(Paiza.Widget widget);
  222. 	}
  223.  
  224. 	public static Paiza getInstance() throws java.lang.Exception {
  225. 		if (paiza == null) {
  226. 			paiza = new Paiza();
  227. 		}
  228. 		return paiza;
  229. 	}	
  230.  
  231. 	public void resolve(Resolver resolver) {
  232. 		init(resolver);
  233. 		for (Widget widget : widgets) {
  234. 			answer(resolver.resolve(widget));
  235. 		}
  236. 		flush();
  237. 	}
  238.  
  239.  
  240. 	public final class Home
  241. 	{
  242. 		private final int H;
  243. 		private final int W;
  244. 		private int[][] display;
  245. 		private Home(int H, int W) {
  246. 			this.H = H;
  247. 			this.W = W;
  248. 			display = new int[H][W];
  249. 		}
  250.  
  251. 		private void setDisplay(int x, int y, int e) throws java.lang.Exception {
  252. 			if (x < 0 || x >= W) {
  253. 				throw new ArrayIndexOutOfBoundsException("x : " + x);
  254. 			}
  255. 			if (y < 0 || y >= H) {
  256. 				throw new ArrayIndexOutOfBoundsException("y : " + y);
  257. 			}
  258. 			if (e != 0 && e != 1) {
  259. 				throw new IllegalArgumentException("e");
  260. 			}
  261. 			display[y][x] = e;
  262. 		}
  263.  
  264. 		public int getH() {
  265. 			return H;
  266. 		}
  267.  
  268. 		public int getW() {
  269. 			return W;
  270. 		}
  271.  
  272. 		public boolean isSpace(int x, int y) {
  273. 			if (x < 0 || y < 0 || x >= W || y >= H) {
  274. 				return false;
  275. 			}
  276. 			return display[y][x] == 0;
  277. 		}
  278. 	}
  279.  
  280. 	public final class Widget
  281. 	{
  282. 		private final int s;
  283. 		private final int t;
  284. 		private Widget(int s, int t) {
  285. 			this.s = s;
  286. 			this.t = t;
  287. 		}
  288.  
  289. 		public int getS() {
  290. 			return s;
  291. 		}
  292.  
  293. 		public int getT() {
  294. 			return t;
  295. 		}
  296. 	}
  297.  
  298. 	private static Paiza paiza = null;
  299.  
  300. 	private Home home;
  301. 	private ArrayList<Widget> widgets;
  302.  
  303. 	private Paiza() throws java.lang.Exception {
  304. 		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
  305. 		String[] hw = in.readLine().split(" ");
  306. 		home = new Home(Integer.parseInt(hw[0]), Integer.parseInt(hw[1]));
  307. 		for (int y = 0; y < home.getH(); y++) {
  308. 			String line = in.readLine();
  309. 			for (int x = 0; x < home.getW(); x++) {
  310. 				home.setDisplay(x, y, (int)(line.charAt(x) - '0'));
  311. 			}
  312. 		}
  313. 		int N = Integer.parseInt(in.readLine());
  314. 		widgets = new ArrayList<Widget>(N);
  315. 		for (int i = 0; i< N; i++) {
  316. 			String[] st = in.readLine().split(" ");
  317. 			widgets.add(new Widget(Integer.parseInt(st[0]), Integer.parseInt(st[1])));
  318. 		}
  319. 	}
  320.  
  321. 	private StringBuilder output = null;
  322. 	private static final String NEWLINE = System.getProperty("line.separator");
  323.  
  324. 	private void init(Resolver resolver) {
  325. 		resolver.setHome(home);
  326. 		output = new StringBuilder(widgets.size() * 6);
  327. 	}
  328.  
  329. 	private void answer(int count) {
  330. 		output.append(count);
  331. 		output.append(NEWLINE);
  332. 	}
  333.  
  334. 	private void flush() {
  335. 		System.out.print(output);
  336. 	}
  337. }
  338.  
Success #stdin #stdout 0.07s 380224KB
comments (?)
stdin
copy 
5 5
00000
00100
00010
10001
10000
3
2 2
1 1
3 2
stdout
copy 
6
20
2
https://ideone.com/d1YjMG
language:
Java (HotSpot 12)
created:
10 years ago
visibility:
secret

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!