CodeForces 514D R2D2 and Droid Army

In the first test the second, third and fourth droids will be destroyed.

In the second test the first and second droids will be destroyed.

题意：有n个机器人，每个机器人有m个标识（用来表示属性），R2D2这个武器最多能打K次，每打一次，所有机器人相对应的属性就要减1，问至少要打几次，能使得连续的机器人属性全部变为0的长度最大，将每个属性打的枪数全部输出来。

思路：很明显这道题要用到线段树，用来记录连续区域间每个属性的最大值（即在这个区间内要把所有的机器人的属性全部变为0所需要打的次数），为了寻找那个最长的连续区域，我们需要用到二分，避免超时。

代码如下：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define Mid(x,y) (x+y)>>1
const int maxn=1e5+10;
int a[maxn][6], ans[6], temp[6];
__int64 k;
int m,n;

struct node
{
	int left, right, val[6];
}rob[4*maxn];

void Build(int l, int r, int rt)
{
	rob[rt].left=l;
	rob[rt].right=r;	
	if(l==r)
	{
		for(int i=1; i<=m; i++)
		{
			rob[rt].val[i]=a[l][i];
		}
		return;
	}
	int mid=Mid(l,r);
	Build(l, mid, L(rt));
	Build(mid+1, r, R(rt));
	for(int i=1; i<=m; i++)
	{
		rob[rt].val[i]=max(rob[L(rt)].val[i], rob[R(rt)].val[i]);
	}
}

int Query(int l, int r, int rt, int x)
{
	if(l==rob[rt].left && r==rob[rt].right)
	{
		return rob[rt].val[x];
	}
	int Max=0;
	if(l<=rob[L(rt)].right)
	{
		if(r<=rob[L(rt)].right)
		{
			Max=Query(l,r,L(rt), x);
		}
		else
		{
			Max=Query(l,rob[L(rt)].right, L(rt), x);
		}
	}
	if(r>=rob[R(rt)].left)
	{
		if(l>=rob[R(rt)].left)
		{
			Max=Query(l,r,R(rt),x);
		}
		else Max=max(Max, Query(rob[R(rt)].left, r, R(rt), x));
	}
	return Max;
}

bool OK(int mid)
{
	bool flag=false;
	for(int i=1; i<=n-mid+1; i++)
	{
		int num=0;
		for(int j=1; j<=m; j++)
		{
			temp[j]=Query(i, i+mid-1, 1, j);
			num+=temp[j];
		}
		if(num<=k)
		{
			flag=true;
			memcpy(ans,temp,sizeof(temp));
			break;
		}
		fill(temp, temp+6, 0);
	}
	if(flag) return true;
	else return false;
}

int main()
{
	while(scanf("%d%d%I64d", &n, &m, &k)!=EOF)
	{
		for(int i=1; i<=n; i++)
		{
			for(int j=1; j<=m; j++)
			{
				scanf("%d", &a[i][j]);
			}
		}
		Build(1,n,1);
	    int le=1, ri=n;
	    while(le<=ri)
	    {
	    	int mid=(le+ri)/2;
	    	if(OK(mid))
			{
				le=mid+1;
			}
			else ri=mid-1;
	    }
	   // printf("%d", ans[1]);
	    for(int i=1; i<m; i++)
	    {
	    	printf("%d ", ans[i]);
	    }
	    printf("%d\n", ans[m]);
	}
	return 0;
}