A
题意:
有n(a0...an-1)个人可以分为k组,求出各组f(l,r)值的和的最小值。
题解:
我们可以将序列分为k组,设我们有k-1个隔板,可以使序列分为k组,当隔板按两数的差值从大到小的顺序插入序列的间隙中时,可以得到f(l,r)各组的值的和的最小值。
存储差值时,可以使用vector来存储,用sort函数对其从大到小排序,每插入一个隔板就把插入隔板处对应的差值从中删除,直到隔板用完。
我们只需要把除插入了隔板的差值外的差值相加即可得到答案。
(所有的差值即两数相减求绝对值(求绝对值的函数abs() ))。
代码:
#include<iostream>
#include<vector>
#include<algorithm>
#include<math.h>
using namespace std;
vector<int>m;
int a[501], s[501];
int n, k, t;
int judge(int x)
{
for (int i = 0; i < k - 1; i++)
{
if (m[i] == x)
{
m.erase(m.begin() + i);
k--;
return 1;
}
}
return 0;
}
int main()
{
cin >> t;
for (int i = 0; i < t; i++)
{
cin >> n >> k;
for (int j = 0; j < n; j++)
{
cin >> a[j];
if (j > 0)m.push_back(abs(a[j] - a[j - 1]));
}
sort(m.rbegin(), m.rend());
for (int l = 1; l < n; l++)
{
if (k > 0 && judge(abs(a[l] - a[l - 1])))
continue;
else
s[i] += abs(a[l] - a[l - 1]);
}
m.clear();
}
for (int i = 0; i < t; i++)
cout << s[i] << endl;
return 0;
}
当然,我们也可以先将所有的差值相加求和,然后再减去差值中前第k-1大之前的所有的数。
代码:
#include<iostream>
#include<vector>
#include<algorithm>
#include<math.h>
using namespace std;
vector<int>m;
int a[501], s[501];
int n, k, t;
int main()
{
cin >> t;
for (int i = 0; i < t; i++)
{
cin >> n >> k;
for (int j = 0; j < n; j++)
{
cin >> a[j];
if (j > 0)m.push_back(abs(a[j] - a[j - 1]));
}
sort(m.rbegin(), m.rend());
for (int l = 1; l < n; l++)
{
s[i] += abs(a[l] - a[l - 1]);
}
for (int r = 0; r < k - 1; r++)
{
s[i] -= m[r];
}
m.clear();
}
for (int i = 0; i < t; i++)
cout << s[i] << endl;
return 0;
}