递增三元组C++

凌窍

于 2024-03-19 16:39:19 发布

阅读量333

点赞数 5

分类专栏：蓝桥杯辅导课之模拟、枚举和排序文章标签： c++ 开发语言

本文链接：https://blog.csdn.net/m0_68416190/article/details/136837862

版权

蓝桥杯辅导课之模拟、枚举和排序专栏收录该内容

5 篇文章 0 订阅

订阅专栏

文章讲述了两种方法来计算满足Ai<Bi<Ci的三元组组合数，一种是使用两个指针遍历已排序的A和C数组，另一种是利用前缀和记录A和C数组中小于等于特定值的元素数量。

摘要由CSDN通过智能技术生成

题目

输入样例：

输出样例：

思路1

题目让我们求所有满足Ai < Bi < Ci的三元组组合，我们可以先将A, B, C三个数组按升序排序，设计分别指向A数组和C数组的指针ia、ic，一开始指向数组的第一个数。对于B数组中的每一个B[i]，利用指针ia，从A数组中找到最后一个小于B[i]的数；利用指针ic，从C数组中找到第一个大于B[i]的数，那么对于特定的一个B[i]，共可以组成ia * 1 * (n - ic)个三元组。随着i的增加，B[i]也会增大，ia、ic只能往后移动……剩下的看代码吧。

代码1

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int a[N], b[N], c[N];
int n;
void input(int arr[])
{
  for (int i = 1; i <= n; i ++)
  cin >> arr[i];
  sort(arr + 1, arr + n + 1);
}

int main()
{
  cin >> n;
  input(a);
  input(b);
  input(c);

  //三元组总个数可能爆int
  LL ia = 0, ic = 0, res = 0;
  
  for (int i = 1; i <= n; i ++)
  {
    while (ia < n && a[ia + 1] < b[i]) ia ++;
    while (ic < n && c[ic + 1] <= b[i])ic ++;
    res += (ia * (n - ic));
  }
  cout << res;
  return 0;
  
}

思路2

前缀和可以表示一个数组前几项的和。如果数组cnt[i]的含义表示A数组中值为i的个数，那么求cnt的前缀和s[i]，s[i]表示A数组中小于等于i的个数，将i替换成B[i]，则s[ B[i] ]表示A数组中小于等于B [i] 的个数。

代码2

/*
利用前缀和 记录a/c数组中<= 某个数有多少个
*/
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

int b[N];
int cnta[N], cntc[N];

int main()
{
  int n, x;
  cin >> n;
  
  for (int i = 1; i <= n; i ++)
  {
    cin >> x;
    //x的最小值为0，为了方便计算前缀和，将x加一
    cnta[++x]++;
  }
  
  //求s[i]
  for (int i = 1; i <= N; i ++)
  cnta[i] += cnta[i - 1];
  
  for (int i = 1; i <= n; i ++)
  {
    cin >> b[i];
    b[i] ++;
  }
  
  for (int i = 1; i <= n; i ++)
  {
    cin >> x;
    cntc[++x]++;
  }
  
  for (int i = 1; i <= N; i ++)
  cntc[i] += cntc[i - 1];
  
  long long res = 0;
  for (int i = 1; i <= n; i ++)
  {
    res += ((long long)cnta[b[i] - 1] * (cntc[N] - cntc[b[i]]));
  }
  
  cout << res;
  return 0;
}