http://codeforces.com/contest/514
n个机器人,每个机器人有m个属性,有k发子弹,每发子弹可以使所有机器人的某个属性减一。当某个机器人的m个属性都变为0时,该机器人视为被摧毁。
求在某段连续区间内消灭机器人数目最多的条件下,机器人的每个属性所需要的子弹数
先二分枚举最长的连续区间,然后根据所得的区间长度,求出某个可行解
O( mn(logn)^2 )
265 ms 49324 KB
#include<bits/stdc++.h>
const int maxn=1e5+10;
using namespace std;
int n,m,K;
int dp[6][maxn][20];
int a[maxn][6];
int res[6],fres[6];
void RMQ(){
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
dp[j][i][0]=a[i][j];
}
}
for(int j=1;(1<<j)<=n;++j){
for(int i=1;i+(1<<j)-1<=n;++i){
for(int k=1;k<=m;++k){
dp[k][i][j]=max(dp[k][i][j-1],dp[k][i+(1<<(j-1))][j-1]);
}
}
}
}
int query(int i,int L,int R){
int k=0;
while((1<<(k+1))<=R-L+1) k++;
return max(dp[i][L][k],dp[i][R-(1<<k)+1][k]);
}
bool ok(int x){
for(int p=1;p+x-1<=n;++p){
int sum=0;
for(int i=1;i<=m;++i){
fres[i]=query(i,p,p+x-1);
sum+=fres[i];
}
if(sum<=K){
for(int i=1;i<=m;++i) res[i]=fres[i];
return true;
}
}
return false;
}
int main()
{
scanf("%d%d%d",&n,&m,&K);
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
scanf("%d",&a[i][j]);
}
}
RMQ();
int l=0,r=n;
while(l<r){
int mid=(l+r)>>1;
if(ok(mid)) l=mid+1;
else r=mid-1;
}
for(int i=1;i<=m;++i){
printf("%d ",res[i]);
}
return 0;
}
#include<bits/stdc++.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=1e5+10;
using namespace std;
int n,m,k;
int mx[6][maxn<<2];
int res[6];
void push_up(int rt){
for(int i=1;i<=m;++i){
mx[i][rt]=max(mx[i][rt<<1],mx[i][rt<<1|1]);
}
}
void build(int l,int r,int rt){
if(l==r){
for(int i=1;i<=m;++i)
scanf("%d",&mx[i][rt]);
return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
push_up(rt);
}
int query(int i,int L,int R,int l,int r,int rt){
if(L<=l&&r<=R) return mx[i][rt];
int m=(l+r)>>1;
int maxx=-1;
if(L<=m) maxx=max(maxx,query(i,L,R,lson));
if(m<R) maxx=max(maxx,query(i,L,R,rson));
return maxx;
}
bool ok(int len){
for(int p=1;p+len-1<=n;++p){
int sum=0;
for(int i=1;i<=m;++i){
sum+=query(i,p,p+len-1,1,n,1);
}
if(sum<=k) return true;
}
return false;
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
build(1,n,1);
int l=0,r=n+1;
while(r-l>1){
int mid=(l+r)>>1;
if(ok(mid)) l=mid;
else r=mid;
}
for(int p=1;p+l-1<=n&&l;++p){
int sum=0;
for(int i=1;i<=m;++i){
int tmp=query(i,p,p+l-1,1,n,1);
res[i]=tmp;
sum+=tmp;
}
if(sum<=k) break;
}
for(int i=1;i<=m;++i){
printf("%d ",res[i]);
}
printf("\n");
return 0;
}