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

Hanks_o發表於2018-04-11

題目傳送門

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

暴力列舉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;
}

相關文章