【ABC321D】题解

[ABC321D] Set Menu 题解

题意简述

给定 $N$ 个整数 $a_1\sim a_n$ 和 $M$ 个整数 $b_1\sim b_m$ 还有一个常数 $P$，每两个整数的和是 $min\{sum,P\}$，求出所有数之和。

思路

对于每个 $a_i$，可以在 $b$ 数组里面找到一个点 $p$：$p$ 之前的数和 $a_i$ 之和小于 $P$，$p$ 及 $p$ 以后的数和 $a_i$ 之和大于等于 $P$。

但是暴力找肯定不行。所以我们可以先将 $b$ 数组从小到大排序，然后二分求解答案。

Accepted Code

Accepted Record

#include<iostream>
#include<algorithm>

using namespace std;

const int MAXN = 2e5 + 10;

int n,m,p,a[MAXN],b[MAXN];
long long sum[MAXN];

int main(){
  cin >> n >> m >> p;
  for (int i = 1; i <= n; i++){
    cin >> a[i];
  }
  for (int i = 1; i <= m; i++){
    cin >> b[i];
  }
  sort(b + 1,b + m + 1);
  for (int i = 1; i <= m; i++){
    sum[i] = sum[i - 1] + b[i];
  }
  long long ans = 0;
  for (int i = 1; i <= n; i++){
    int x = lower_bound(b + 1,b + m + 1,p - a[i] + 1) - b - 1;
    ans += 1ll * a[i] * x + sum[x] + 1ll * p * (m - x);
  }
  cout << ans;
  return 0;
}