Hdu 6534 Chika and Friendly Pairs 莫隊算法+樹狀數組

題目鏈接


題意求給區間[L,R]中有少對(i,j)滿足i<j且abs(a[i]-a[j])<=k.

首先來說暴力的方法就是離散化,然後用樹狀數組來維護,但是m次詢問,m很大,所以說一定會t。

這裏用到了莫對算法分塊來降低複雜度,莫隊算法的核心思想是對所有查詢進行合理的排序,並將本次查詢的結果不斷進行修改以得到下次查詢的結果

AC代碼:

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2]={0,1,0,-1,1,0,-1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=3e4+5;
int lt[maxn],mt[maxn],rt[maxn];
int a[maxn],b[maxn],c[maxn],be[maxn],len;
int n,m,k,ans,u;
int res[maxn];
struct node
{
    int l,r,id;
}s[maxn];
bool cmp(node s,node e)
{
    if(be[s.l]==be[e.l]) return s.r<e.r;
    return s.l<e.l;
}
void update(int x,int val)
{
    while(x<=len)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    int p=0;
    while(x>0)
    {
        p+=c[x];
        x-=lowbit(x);
    }
    return p;
}
void add(int x)
{
    ans+=(getsum(rt[x])-getsum(lt[x]));
    update(mt[x],1);
}
void del(int x)
{
    update(mt[x],-1);
    ans-=(getsum(rt[x])-getsum(lt[x]));
}
int main()
{
    int l,r;
    memset(c,0,sizeof(c));
    scanf("%d %d %d",&n,&m,&k);
    u=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[i]=a[i];
        be[i]=i/u+1;//分塊
    }
    sort(b+1,b+1+n);
    len=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++)
    {
        lt[i]=lower_bound(b+1,b+1+len,a[i]-k)-b-1;//a[i]-k對應離散後的下標-1
        rt[i]=upper_bound(b+1,b+1+len,a[i]+k)-b-1;//a[i]+k對應離散後的下標
        mt[i]=lower_bound(b+1,b+1+len,a[i])-b;//a[i]對應的下標
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d",&l,&r);
        s[i].l=l,s[i].r=r,s[i].id=i;
    }
    sort(s+1,s+1+m,cmp);//分塊排序降低複雜度
    l=1,r=0,ans=0;
    for(int i=1;i<=m;i++)//開始對每一個區間進行處理,移動左右端點
    {
        while(l<s[i].l) del(l++);
        while(l>s[i].l) add(--l);
        while(r>s[i].r) del(r--);
        while(r<s[i].r) add(++r);
        res[s[i].id]=ans;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",res[i]);
    return 0;
}

莫隊算法思想:https://blog.csdn.net/ThinFatty/article/details/72581276

發佈了97 篇原創文章 · 獲贊 98 · 訪問量 6575
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章