結果

問題 No.918 LISGRID
コンテスト
ユーザー 37zigen
提出日時 2020-03-11 14:07:52
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 1,564 bytes
コンパイル時間 3,179 ms
コンパイル使用メモリ 83,620 KB
実行使用メモリ 60,868 KB
最終ジャッジ日時 2024-11-16 00:10:45
合計ジャッジ時間 22,830 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 5 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Scanner;

class Main {
	public static void main(String[] args) {
		new Main().run();
	}
	
	void run() {
		Scanner sc=new Scanner(System.in);
		int H=sc.nextInt();
		int W=sc.nextInt();
		int[] A=new int[H];
		int[] B=new int[W];
		int[][] m=new int[H][W];
		for(int h=0;h<H;++h)for(int w=0;w<W;++w)m[h][w]=-1;
		for(int i=0;i<A.length;++i) {
			A[i]=-sc.nextInt();
		}
		for(int i=0;i<B.length;++i) {
			B[i]=-sc.nextInt();
		}
		Arrays.sort(A);
		Arrays.sort(B);
		for(int i=0;i<A.length;++i)A[i]*=-1;
		for(int i=0;i<B.length;++i)B[i]*=-1;
		int s=1,t=H*W;
		for(int h=0;h<H;++h) {
			int bottom=0,top=0;
			for(int w=0;w<W;++w) {
				if(B[w]>1)++bottom;
				else ++top;
			}
			if(A[h]>1) {
				if(top==0) {
					for(int w=0;w<W;++w) {
						m[h][(w+W-A[h])%W]=s++;
						--B[w];
					}
				}else {
					int left=Math.min(bottom,A[h]-1);
					int right=A[h]-left;
					for(int w=0;w<bottom;++w) {
						m[h][(w+bottom-left)%bottom]=s++;
						--B[w];
					}
					for(int w=0;w<top;++w) {
						m[h][bottom+(w+top-right)%top]=t-(top-1)+w;
					}
					t-=top;
				}
			}else {
				for(int w=0;w<W;++w) {
					for(int i=h;i<H;++i) {
						if(B[w]>1) {
							m[i][w]=s++;
							B[w]--;
						}else m[i][w]=t--;
					}
				}
				break;
			}
		}
		PrintWriter pw=new PrintWriter(System.out);
		for(int h=0;h<H;++h) {
			for(int w=0;w<W;++w) {
				pw.print(m[h][w]+(w==W-1?"\n":" "));
			}
		}
		pw.close();
	}
	
	void tr(Object...objects) {System.out.println(Arrays.deepToString(objects));}
}
0