算法-区间和

题目一

 解题思路

注意点:代码中对容器进行排序去重操作是针对查询坐标的重复,而非数轴上加减法的操作。

 

该题的一些其他解法https://www.acwing.com/solution/content/10915/

代码模板

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

const int N=300010;
int n,m;
int a[N],s[N];
vector<int>alls ;
vector<pair<int,int>>add,query;

int find(int x)
{
    int l=0,r=alls.size()-1;
    while(l<r)
    {
        int mid=l+r>>1;
        if(alls[mid]>=x)r=mid;
        else l=mid+1;
        
    }
    return r+1;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int x,c;
        scanf("%d%d",&x,&c);
        add.push_back({x,c});
        alls.push_back(x);
    }
    for(int i=1;i<=m;i++)
    {
        int l,r;
        scanf("%d%d",&l,&r);
        query.push_back({l,r});
        alls.push_back(l);
        alls.push_back(r);
    }
    sort(alls.begin(),alls.end());
    alls.erase(unique(alls.begin(),alls.end()),alls.end());
    
    for(auto item:add){
        int x=find(item.first);
        a[x]+=item.second;
    }
    for(int i=1;i<=alls.size();i++)s[i]=s[i-1]+a[i];
    for(auto item:query)
    {
        int l=find(item.first);
        int r=find(item.second);
        printf("%d\n",s[r]-s[l-1]);
    }
    return 0;
}

自己写的超时代码(结果正确)

利用前缀和算法改进后可通过检测

#include<iostream>
using namespace std;
const int N=1e5+10;
int q[N][2];
int n,m;
void quicksort(int l,int r)
{
    if(l>=r)return;
    int i=l-1,j=r+1;
    int x=q[i+j+1>>1][0];
    while(i<j)
    {
        do i++;while(q[i][0]<x);
        do j--;while(q[j][0]>x);
        if(i<j)
        {
            swap(q[i][0],q[j][0]);
            swap(q[i][1],q[j][1]);
        }
    }
    quicksort(l,i-1);
    quicksort(i,r);
}

int find(int l)//寻找映射数组的匹配左边界
{
    int i=0,j=n-1;
    int mid=i+j+1>>1;
    while(i<j)
    {
        int mid =i+j>>1;
        if(q[mid][0]<l)i=mid+1;
        else j=mid;
    }
    if(q[i][0]<l)
    return i+1;
    else
    return i;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d%d",&q[i][0],&q[i][1]);
    quicksort(0,n-1);
    while(m--)
    {
        int l,r,sum=0;
        scanf("%d%d",&l,&r);
        for(int i=find(l);q[i][0]<=r&&i<n;i++)
        {
            sum+=q[i][1];
        }
        printf("%d\n",sum);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xx_xb/article/details/132193634