D. Inconvenient Pairs(排列组合？+思维

不吃土司边

于 2021-09-10 14:31:19 发布

阅读量241

点赞数

分类专栏： acm 文章标签： c语言算法

本文链接：https://blog.csdn.net/weixin_45714303/article/details/120221482

版权

acm 专栏收录该内容

302 篇文章 1 订阅

订阅专栏

本文介绍了一种使用C++实现的解决方案，针对给定的数据集，通过区间覆盖问题来处理在两个有序数组x和y中，根据特定查询k，计算满足条件p[i].first <= x[i]且p[i].second >= y[i]的区间个数。文章涉及了动态规划和区间查询的技巧，并展示了如何在两种情况下（按x和y排序）进行优化。

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

#include<bits/stdc++.h>
using namespace std;
int t,n,m,k;
const int N=3e5+10;
int x[N],y[N];
pair<int,int> p[N];
bool cmp(pair<int,int> a,pair<int,int> b){
    return a.second<b.second;
}
map<int,int> mp;
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&m,&k);
        for(int i=1;i<=n;i++) scanf("%d",&x[i]);
        for(int i=1;i<=m;i++) scanf("%d",&y[i]);
        for(int i=1;i<=k;i++) scanf("%d%d",&p[i].first,&p[i].second);
        sort(p+1,p+1+k);
        int now=1; long long ans1=0,tp=0,sum=0;
        for(int i=2;i<=n;i++){
            mp.clear();
            int l=x[i-1],r=x[i];
            while(p[now].first==l&&now<=k) ++now;
            while(p[now].first<r&&now<=k) mp[p[now].second]++,now++;
            for(auto it:mp){
                sum+=((it).second);
                tp+=(((it).second)-1)*((it).second)/2;
            }
            ans1+=(sum-1)*sum/2-tp
            tp=sum=0;
        }
//        cout<<ans1<<endl;
        sort(p+1,p+1+k,cmp);
        now=1;tp=sum=0;
        for(int i=2;i<=m;i++){
            mp.clear();
            int l=y[i-1],r=y[i];
            while(p[now].second==l&&now<=k) ++now;
            while(p[now].second<r&&now<=k) mp[p[now].first]++,now++;
            for(auto it:mp){
                sum+=((it).second);
                tp+=(((it).second)-1)*((it).second)/2;
            }
            ans1+=(sum-1)*sum/2-tp;
            tp=sum=0;
        }
        cout<<ans1<<endl;
    }
}