1. 程式人生 > >bzoj5177: [Jsoi2013]貪心的導遊(主席樹)

bzoj5177: [Jsoi2013]貪心的導遊(主席樹)

題目傳送門

解法:
想了半天只有暴力。
結果還真是暴力。。

暴力列舉p的倍數。
用主席樹找出區間內p*i~p*(i+1)-1的最大數。
然後減去P*i。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
struct node {int
lc,rc,c;}t[11000000];int cnt,rt[1100000]; void build(int &u,int l,int r,int p) { if(u==0)u=++cnt;t[u].c++; if(l==r)return ;int mid=(l+r)/2; if(p<=mid)build(t[u].lc,l,mid,p); else build(t[u].rc,mid+1,r,p); } void Merge(int &u1,int u2) { if(u1==0) {u1=u2;return ;}if(u2==0) return ; t[u1].c+=t[u2].c; Merge(t[u1].lc
,t[u2].lc); Merge(t[u1].rc,t[u2].rc); } int find(int u1,int u2,int l,int r,int st,int ed) { if(t[u1].c-t[u2].c==0)return 0; if(l==r) return l;int mid=(l+r)/2; if(ed<=mid)return find(t[u1].lc,t[u2].lc,l,mid,st,ed); if(st>mid)return find(t[u1].rc,t[u2].rc,mid+1,r,st,ed); int
c=t[t[u1].rc].c-t[t[u2].rc].c; if(c>0) { int sum=find(t[u1].rc,t[u2].rc,mid+1,r,st,ed); if(sum==0)return find(t[u1].lc,t[u2].lc,l,mid,st,ed); } else find(t[u1].lc,t[u2].lc,l,mid,st,ed); } int main() { int n,m;scanf("%d%d",&n,&m);cnt=0; for(int i=1;i<=n;i++) { int x;scanf("%d",&x); build(rt[i],0,1000,x);Merge(rt[i],rt[i-1]); } for(int i=1;i<=m;i++) { int l,r,p;scanf("%d%d%d",&l,&r,&p);l++;r++; int ans=0; for(int j=0;j<=1000/p;j++) { ans=max(ans,find(rt[r],rt[l-1],0,1000,p*j,p*(j+1)-1)-p*j); }printf("%d\n",ans); } return 0; }