题目https://www.acwing.com/problem/content/description/149/
题意:有n+1栋楼,每相邻两两配对搭一条电缆,现资源k条电缆,问选哪k对消耗最小。(一栋楼不难同时被两栋或以上的其他楼相连)
数据范围
2≤n≤100000,
1≤k≤n/2,
0≤s≤1000000000
输入样例:
5 2
1
3
4
6
12
输出样例:
4
思路:这道题目我们首先很容易发现性质,最优解中每两个配对的办公楼一定是相邻的,既然如此的话不妨设置一个数组D,D[i]表示第i个办公楼和第i-1个办公楼之前的距离.那么我们就可以用堆和链表来解决。
代码实现:
#include<iostream>
#include<cstring>
#include<queue>
#include<vector>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<ll, int> PLI;
const int N = 1e5 + 5;
int n, k;
int l[N], r[N];
ll d[N];
//从双链表里删除一个数
void delete_node(int p)
{
r[l[p]] = r[p];
l[r[p]] = l[p];
}
int main()
{
cin >> n >> k;
for(int i = 0; i < n; i ++ )
cin >> d[i];
for(int i = n - 1; ~i; i -- )
d[i] -= d[i - 1];
set<PLI> s;
d[0] = d[n] = 1e15;
for(int i =0; i <= n; i ++ ){
l[i] = i - 1;
r[i] = i + 1;
if(i >= 1 && i < n)
s.insert({d[i], i});
}
ll res = 0;
while(k -- ){
//取出里面的最小值
set<PLI>::iterator it = s.begin();
ll v = it -> first;
int p = it -> second;
int left = l[p], right = r[p];
//如果这个一对取用了,那么它的左右两点不能再用,必须从set和链表中删除
s.erase(it);
s.erase({d[left], left});
s.erase({d[right], right});
delete_node(left);
delete_node(right);
res += v;
d[p] = d[left] + d[right] - d[p];
s.insert({d[p], p});
}
cout << res << endl;
return 0;
}