Програмчлалын XV онлайн тэмцээний "Чөтгөрийн тоо" бодлогын бодолт

bodolt

var a,a1,a2:array[1..20]of int64;
n,k,i:integer;
begin
read(n,k);
if k>6 then
begin
a[1]:=k-2;
a1[1]:=1;
a2[1]:=0;
a[2]:=(k-1)*(k-1);
a1[2]:=k-2;
a2[2]:=1;
for i:=3 to n do
begin
a[i]:=(k-1)*(a1[i-1]+a2[i-1]+a[i-1]);
a1[i]:=a[i-1];
a2[i]:=a[i-2];
end;
writeln(a[n]+a1[n]+a2[n]);
end
else
begin
a[1]:=k-1;
for i:=2 to n do
a[1]:=a[1]*k;
writeln(a[1]);
end;
readln;
end.