f=1:scanl(+)1 f
main=getContents>>=mapM(print.sum.map(^2).flip take f.read).tail.lines