难度不高，容易想到对朴素 $dp$ 的 $j$ 分层，线段树的优化也比较显然。