Hdu 6534 Chika and Friendly Pairs 莫队算法+树状数组
生活随笔
收集整理的這篇文章主要介紹了
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對應離散后的下標-1rt[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
總結
以上是生活随笔為你收集整理的Hdu 6534 Chika and Friendly Pairs 莫队算法+树状数组的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: R19436221 区间第k小 主席树
- 下一篇: XOR and Favorite Num