Удвоить трѐхзначное число,...

Удвоить трѐхзначное число, введенное с клавиатуры, если оно содержит в своей записи хотя бы одну единицу, и возвести в квадрат в другом случае.Если введѐнное число не трѐхзначное, вывести сообщение об ошибке.(В ПАСКАЛЕ)

 

program test;

var
    num, n : longint;
    i : integer;
    f : boolean;

begin
    readln(num);
    n := num;
    f := false;
    i := 0;
    repeat
        if n mod 10 = 1 then
            f := true;
        n := n div 10;
        i := i + 1;
    until n = 0;
    if i <> 3 then begin
        Writeln('Chislo ne treh-znachnoe.');
    end
    else begin
        if f then
            num := num * 2
        else begin
            num := sqr(num); end;
        writeln('Rezultat: ', num)
    end;
end.