X轴上有N个点,每个点除了包括一个位置数据X[i],还包括一个权值W[i]。点P到点P[i]的带权距离 = 实际距离 * P[i]的权值。求X轴上一点使它到这N个点的带权距离之和最小,输出这个最小的带权距离之和。
输入
第1行:点的数量N。(2 <= N <= 10000)
第2 - N + 1行:每行2个数,中间用空格分隔,分别是点的位置及权值。(-10^5 <= X[i] <= 10^5,1 <= W[i] <= 10^5)
输出
输出最小的带权距离之和。
输入样例
5
-1 1
-3 1
0 1
7 1
9 1
输出样例
20
解题思路:
因为最后要乘以权值w[i],那么我们可以想象成有w[i]个x[i]坐标组成的,那么我们又转化为中位数的问题了,首先我们将统计有多少个数(也就是把所有的权值相加),然后除以2,然后按照从小到大排序,找最中间那个数,找到记录那个点的坐标x[i],然后剩下的就简单了,按照题目意思做就行了。
#include <bits/stdc++.h> using namespace std; const int maxn = 1e5 + 5; typedef long long ll; struct node { int x, p; }a[maxn]; int n; bool cmp (node a, node b) { return a.x < b.x; } ll get(int p) { ll sum = 0; for (int i = 1; i <= n; i++) sum += 1LL * abs(a[i].x - a[p].x) * a[i].p; return sum; } int main() { int p1, p2; scanf("%d", &n); ll sum = 0; for (int i = 1; i <= n; i++) { scanf("%d%d", &a[i].x, &a[i].p); sum += a[i].p; } sort (a + 1, a + n + 1, cmp); ll x1 = sum / 2, x2 = sum / 2 + 1, ans = 1e18; for (int i = 1; i <= n; i++) { if (x1 > 0 && x1 - a[i].p <= 0) p1 = i; if (x2 > 0 && x2 - a[i].p <= 0) p2 = i; x1 -= a[i].p; x2 -= a[i].p; } ans = min(ans, get(p1)); ans = min(ans, get(p2)); printf("%lld\n", ans); return 0; }