Из одномерного массива в матрицу. Java

[Astar]

Задача: Из строки в матрицу. Элементы одномерного массива А(n^2) построчно расположить в матрице В (n,n).
Проблема в том, что программу нужно написать на java, на Паскале получилось нечто подобное
( но размерность массива опять же задана изначально... )

uses crt;

var a:array[1..n*n]of integer;
b:array[1..n,1..n]of integer;
i,j:integer;

begin
clrscr;
write('Введите массив А -> ');
for i:=1 to sqr(n) do read(a[i]);
readln;

for i:=1 to n do for j:=1 to n do
b[i,j]:=a[i+(j-1)*n];
...
end.

на Яве дело дошло лишь до этого -

import java.util.*;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class nj4
{
public static void main(String args[])throws IOException {
{
BufferedReader input = new BufferedReader(
new InputStreamReader(System.in));
System.out.print("Enter size:\n> ");
int i, size = Integer.parseInt(input.readLine());
double[]array = new double[size];
System.out.println("Enter array:");
for (i = 0; i < size; ++i)
array[i] = Double.parseDouble(input.readLine() );

...
}
}
}

пользователь вводит размер массива и его значение, а как после этого извернуться и преобразовать его в матрицу (по сути одномерный в двумерный массив) для меня несколько проблематично...
Помогите, кто чем может))