贪心
先转化一下题意
首先如果一个人的家和办公室在河同一侧那么建桥的时候不用去考虑它,最终把答案加上即可
在河两侧的家和办公室互换不影响答案,那么可以把这个抽象到一个区间$[l,r]$,距离就是$|l-x|+|r-x|$
如果$k=1$时,直接取中位数即可
那么考虑$k=2$时,对于某一个人来说,它对答案的贡献就是$min(|l-x_1|+|r-x_1|,|l-x_2|+|r-x_2|)$,$x_1,x_2$为修建两个桥的位置
但是如何快速判断某一个人要走哪个桥,首先如果有一座桥在区间内,那么一定走那座桥,如果没有,那么就走离最近端点最近的那座桥,更本质地讲,就是找离区间中点最近的桥
那么可以根据区间中点将区间排序,然后枚举断点,将左右两端各视作$k=1$时的子问题
现在考虑如何维护一个集合的中位数
发现如果加入或删除某一个数,中位数最多移动$1$个数,那么只需要用set找出每一个当前在序列中的数的前驱后继即可