昨天在UOJ群里看这个问题，今天就会了 $O(n)$ 做法。可惜洛谷博客太小，写不下。