[ABC371D] 1D Country 题解

合集 - 题解(15)

1.[ABC347C] Ideal Holidays题解2024-03-312.《扶苏的问题》题解2024-06-023.[ABC366C] Balls and Bag Query 题解2024-08-124.[ABC366D] Cuboid Sum Query 题解2024-08-125.P7706 文文的摄影布置 题解2024-08-216.[ABC370C] Word Ladder 题解2024-09-087.P11020 「LAOI-6」Radiation 题解2024-09-088.P11019 「LAOI-6」[太阳]] 请使用最新版手机 QQ 体验新功能 题解2024-09-08

9.[ABC371D] 1D Country 题解2024-09-15

10.[ABC371D] 1D Country 线段树解法2024-09-1511.P1955 程序自动分析 题解2024-10-1812.[ABC376E] Max × Sum 题解2024-10-2013.[ABC379D] Home Garden 题解2024-11-1014.P11276 第一首歌 题解2024-11-1515.[ABC380C] Move Segment 题解2024-11-18

收起

这题，怎么说呢， $STL$ 大法好。

前置芝士： lower_pound 函数在结构体上的使用。

那其实这题便是一个二分前缀和的水题了。结构体存储每个村庄的距离 $x$ ，人口 $d$ 。对于每个输入的 $[l,r]$ 二分查找其对应的村庄，进行一次答案的统计，输出即可。

代码 :

#include <bits/stdc++.h>
#define seq(q, w, e) for (int q = w; q <= e; q++)
#define ll long long
using namespace std;
const int maxn = 2e5+10;
ll n,m,ans;
struct node{
    ll x,d;
    node(){}
    node(int a,int b):x(a),d(b){}           //初始化
    bool operator < (const node b)const{    //自定义比较函数
        return x<b.x;
    }
}a[maxn];                                   //村庄数组

bool cmp(node a,node b){
    return a.x<b.x;                         //排序函数，以距离为关键词
}

signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n;
    a[0].d=0;
    seq(i,1,n){
        cin>>a[i].x;
    }
    seq(i,1,n){
        cin>>a[i].d;
    } 
    sort(a+1,a+n+1,cmp);                  //输入结构体后按距离排序
    seq(i,1,n){
        a[i].d+=a[i-1].d;                 //维护前缀和
    }
    cin>>m;
    seq(i,1,m){
        ans=0;
        int l,r;
        cin>>l>>r;
        int sum1=lower_bound(a+1,a+1+n,node(l,0))-a,sum2=lower_bound(a+1,a+1+n,node(r,0))-a;  //二分查找求区间
        if(a[sum2].x!=r) sum2--;      //由于low_pound返回的是第一个大于等于的数，所以如果不相等，需要减一
        ans=a[sum2].d-a[sum1-1].d;    //前缀和查询操作。
        cout<<ans<<endl;
    }
    return 0;
}