AcWing 1236. 递增三元组 解题思路及代码

先贴个题目：

以及题目链接： 1236. 递增三元组 - AcWing题库https://www.acwing.com/problem/content/1238/

来讲讲思路，第一反应也是暴力枚举，直接三重循环判断，刚准备动手，额O(N立方)的复杂度和N=1e5的现实让我冷静了下来，这肯定是tle，我们先看数据范围，这么极限的数据，连O(n方)都超时而且 O(n)的算法我肯定想不到，三个变量的组合，因此算法的时间复杂度锁定在O(nlogn)这里，那我们只能用一次一重循环，也需可以使用快排来辅助我们操作，经过看题解思考，我们选择使用前缀和的做法，开两个个数组，分别处理a，c两个数组，每个数组的下标代表了小等它的数在数组中的数量，然后开一个数组存b数列，然后枚举b数组，直接找出a数组比他小的数的数量和c数组比他大的数的数量然后相乘，最后全部相加即为答案。

代码如下：

#include <iostream>
using namespace std;

const int N = 1e5 + 10;

int a[N], b[N], c[N];

int main()
{
    int n, num;
    cin >> n;
    for (int i = 1; i < n + 1; ++i)
    {
        scanf("%d",&num);
        a[num]++;
    }
    for (int i = 1; i < n + 1; ++i)
        scanf("%d",&b[i]);
    for (int i = 1; i < n + 1; ++i)
    {
        scanf("%d",&num);
        c[num]++;
    }
    for (int i = 1; i < N; ++i)
    {
        a[i] = a[i - 1] + a[i];
        c[i] = c[i - 1] + c[i];
    }
    long long ans = 0;
    for (int i = 0; i < n + 1; ++i)
    {
        ans += (long long)a[b[i] - 1] * (c[100000] - c[b[i]]);
    }
    cout << ans;
    return 0;
}

提一嘴，ans可能爆int，因为又可能跑到1e10的量级，所以要开long long。

然后，我经过前缀和的解法的启发，根据我的初始思路，能不能用排序解决呢，然后想到了对a,c数组进行排序，枚举b然后二分答案的做法，代码如下：

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 1e5 + 10;

int a[N], b[N], c[N];

int main()
{
    int n;
    cin >> n;
    for (int i = 1; i < n + 1; ++i)
        scanf("%d", &a[i]);
    for (int i = 1; i < n + 1; ++i)
        scanf("%d", &b[i]);
    for (int i = 1; i < n + 1; ++i)
        scanf("%d", &c[i]);
    sort(a + 1, a + n + 1);
    sort(c + 1, c + n + 1);
    long long ans = 0;
    int as, cs;
    for (int i = 1; i < n + 1; ++i)
    {
        int left = 1, right = n;
        while (left < right)
        {
            int mid = (left + right + 1) / 2;
            if (a[mid] >= b[i])
                right = mid - 1;
            else
                left = mid;
        }
        if (a[left] >= b[i])
            as = 0;
        else
            as = left;
        left = 1, right = n;
        while (left < right)
        {
            int mid = (left + right) / 2;
            if (c[mid] > b[i])
                right = mid;
            else
                left = mid + 1;
        }
        if (c[left] <= b[i])
            cs = 0;
        else
            cs = n-left+1;
        ans += (long long)as * cs;
    }
    cout << ans;
    return 0;
}

代码也没啥要注意的，主要是要记得讨论下二分完发现a的答案依旧大于等于b，c的答案依旧小于等于b的情况即可。

by————2024.2.27刷题记录