ll@n,@m,@a[m],@q,z=sum[i,0,m](a[i]**2); rep(q){ ll@c--,@k,@d--; z-=a[c]**2+a[d]**2; a[c]-=k; a[d]+=k; z+=a[c]**2+a[d]**2; wt(z); }