P13页代码的long long ans; 1应该改成long long ans=0;。

P54页代码的第 $3$ 行的if(n - x + y < k)应该改成if(n - x + y + 1 < k),不然过不了P1036（亲测）。