BZOJ3747: [POI2015]Kinoman
线段树
题解:
线段树经典题
nxt[i]记录第i天的电影下次播放时间
枚举区间左端点,线段树维护每个位置作为右端点的答案
考虑l-r的左端点变为l+1
发现l到nxt[l]-1的答案减少w[f[l]]
而nxt[l]到nxt[nxt[l]]-1增加w[f[l]]
线段树维护,支持区间修改以及查询最大值
Code:
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#define D(x) cout<<#x<<" = "<<x<<" "
#define E cout<<endl
using namespace std;
typedef pair<int,int> pii;
const int N = 500005;
int n,K,L,R,a[N],s[N];
long long ans;
struct RMQ{
pii d[N<<1][20];
void init(int a[]){
for(int i=1;i<=n;i++){ d[i][0].first=a[i]; d[i][0].second=i; }
for(int j=1;j<=19;j++){ for(int i=1;i<=n;i++){ d[i][j]=max(d[i][j-1],d[i+(1<<(j-1))][j-1]); } }
}
pii query(int l,int r){
int i=19; while((1<<i) > (r-l+1)) i--;
return max(d[l][i],d[r-(1<<i)+1][i]);
}
} st;
struct Data{
int a,lb,rb,pos,val;
Data(int _a,int _lb,int _rb,int _pos,int _val){ a=_a; lb=_lb; rb=_rb; pos=_pos; val=_val; }
bool operator < (const Data &other) const { return val < other.val; }
void debug(){ puts("Data"); D(a); D(lb); D(rb); D(pos); D(val); E; }
};
priority_queue<Data> q;
void insert(int a,int lb,int rb){
// D(a); D(lb); D(rb); E;
if(rb < lb) return;
pii x=st.query(lb,rb);
// D(x.second); E;
q.push(Data(a,lb,rb,x.second,s[x.second]-s[a-1]));
}
int main(){
freopen("a.in","r",stdin);
scanf("%d%d%d%d",&n,&K,&L,&R);
for(int i=1;i<=n;i++) scanf("%d",a+i);
for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
st.init(s);
for(int i=1;i<=n-L+1;i++) insert(i,i+L-1,min(i+R-1,n));
while(K--){
Data x=q.top(); q.pop();
// x.debug();
ans+=x.val;
insert(x.a,x.lb,x.pos-1);
insert(x.a,x.pos+1,x.rb);
}
printf("%lld\n",ans);
}
本文介绍了解决BZOJ3747:[POI2015]Kinoman问题的方法,利用线段树进行区间修改及查询最大值操作,通过枚举区间左端点,维护每个位置作为右端点的答案。
734

被折叠的 条评论
为什么被折叠?



