（前缀和/思维）Codeforces Round #806 (Div. 4)F. Yet Another Problem About Pairs Satisfying an Inequality

最新推荐文章于 2024-10-08 09:12:18 发布

ZaneBobo

最新推荐文章于 2024-10-08 09:12:18 发布

阅读量173

点赞数 2

分类专栏： cf题解文章标签：大数据 c++ 贪心算法 c语言

本文链接：https://blog.csdn.net/weixin_61904259/article/details/125765854

版权

cf题解专栏收录该内容

13 篇文章 0 订阅

订阅专栏

该博客介绍了如何解决数组中寻找满足ai<i<aj<j的数对个数的问题。通过前缀和思想，建立i<a[j]的关系，并利用限制条件a[i]<i和a[j]<j来计算数对数量。代码实现中使用了C++，并展示了详细的解决步骤和代码逻辑。

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

一.题意

给你一个数组，要求你找出数组中满足 ai<i<aj<j 的数对个数。

二.思路

首先限定数对中的每一个数一定满足ai<i，那么只需要建立一个i<aj的关系即可。

我们利用前缀和的思想 sum[i]代表 i之前包括i 所有满足ai<i的数的个数，那么sum[a[j]-1]就是所有满足i<a[j]的且满足ai<i的个数。然后枚举的时候再加上一个限定条件 a[j]<j 就行啦。

三.代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <map>

using namespace std;
const int N = 2e5+10;
int a[N];
map<int,int> sum;
typedef long long LL;

void solve()
{
    int n;
    cin>>n;
    sum.clear();
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum[i]+=sum[i-1]+(a[i]<i);//限定 a[i]<i
    }
    LL res=0;
    for(int i=1;i<=n;i++)
    {
        if(i>a[i])res+=sum[a[i]-1];
        //if条件限定a[j]<j sum[a[i]-1]则是限定了i<a[j]这样全部符合条件的对数
    }
    cout<<res<<endl;
    
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        solve();
    }
}