题意:有n个男生和m个女生,知道他们的身高。给他们配对,要求所有配对的身高差之和最小。
数据范围:0 < n, m <= 10000, |n – m| <= 100
解析:根据数据范围分析,10^4 * 100正好在合理时间之内。
先给两个序列进行排序,用短的序列去匹配长的序列。设两个序列之差为len,那么短的序列可以与长序列中len+1个元素匹配取最优,记这个长度为跨度。
用dp[i][j]表示短的那个序列去匹配长的序列,i表示短序列的第i个元素,j表示跨度大小。那么转移方程就是:
dp[i][j] = min(dp[i][j-1], dp[i-1][j] + |a[i+j] - b[i]|)(假设a是长的那个序列)
具体见代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <stdlib.h>
using namespace std;
const int N = 10005;
double a[N], b[N];
double dp[2][105];
double solve(double p[], double q[], int l1, int l2){
memset(dp, 0, sizeof(dp));
int len = l1 - l2 + 1;
for(int i = 0; i < l2; i++){
dp[i%2][0] = dp[(i+1)%2][0] + fabs(p[i] - q[i]);
for(int j = 1; j < len; j++)
dp[i%2][j] = min(dp[i%2][j-1], dp[(i+1)%2][j] + fabs(p[i+j]-q[i]));
}
return dp[(l2-1)%2][len-1];
}
int main(){
int n, m;
while(scanf("%d%d", &n, &m) && (n + m)){
for(int i = 0; i < n; i++)
scanf("%lf", &a[i]);
for(int i = 0; i < m; i++)
scanf("%lf", &b[i]);
double ans;
sort(a, a + n);
sort(b, b + m);
if(n > m)
ans = solve(a, b, n, m);
else
ans = solve(b, a, m, n);
printf("%.6lf\n", ans);
}
return 0;
}