【DP 好题】Kick Start 2019 Round C Catch Some

题目链接

题目大意

在一条数轴上住着 $N$ 条狗和一个动物研究者 Bundle。Bundle 的坐标是 0,狗的坐标都是正整数,可能有多条狗住在同一个位置。每条狗都有一个颜色。Bundle 需要观测 $K$ 条狗。要观测一条狗 Bundle 必须走到狗的住处,并且穿着和狗同色的衣服。Bundle 只能在家换衣服。试问 Bundle 至少要走多长的路程?注意:最后 Bundle 不必回到住处。

Constraints

  • 不超过 100 组测试数据
  • $ 1 le N le 1000 $
  • $ 1 le K le N $
  • Time limit: 30 s

分析

若 Bundle 最后必须回到住处,反而更好处理,并且复杂度可以做到 $O(N log C)$,$C$ 表示狗的颜色总数。
思路:从左往右,把相邻的同色狗之间的距离放到一个小顶堆中。

在最后不必回到住处的条件下,貌似就只能 DP 了。

Let dp[i][j][k] denotes the minimum amount of walk needed to observe j dogs such that Bundle has either observed or has decided she'll not observe dogs of colors 1 to i and k is a boolean denoting if we have chosen the last color in the first i colors.

这篇题解提供了另一种思路。

原文地址:https://www.cnblogs.com/Patt/p/11616676.html