Codeforces 514D R2D2 and Droid Army RMQ问题

题目大意：

就是现在对于一个m行n列的矩阵, 每次可以选择m行中的任意一行的正整数-1,最多选择k次, 如果某一列中的m个数全部成为了0, 那么这一列算作被破坏, 求被破坏的连续的列 最长是多少列, 并且输出选择各行的次数 (m <= 5, n <= 10 ^5, k <= 10 ^9, n*m的矩阵中的数为<= 10^8的非负整数)

大致思路：

就是RMQ预处理一下每一行的所有数, 然后二分能够找到的长度判断是否可行即可, 很简单的RMQ问题

代码如下：

Result  :  Accepted     Memory  :  49724 KB     Time  :  217 ms

/*
 * Author: Gatevin
 * Created Time:  2015/2/25 13:00:53
 * File Name: poi~.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

int n, m, K, a[6][100010];
int dp[6][100010][20];
int ans[100010];

void initRMQ()
{
    for(int i = 1; i <= m; i++)
    {
        for(int j = 1; j <= n; j++)
            dp[i][j][0] = a[i][j];
        for(int l = 1; (1 << l) <= n; l++)
            for(int j = 1; (j + (1 << l) - 1) <= n; j++)
                dp[i][j][l] = max(dp[i][j][l - 1], dp[i][j + (1 << (l - 1))][l - 1]);
    }
}

int askRMQ(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 check(int mid)//判断mid是否可行
{
    for(int i = 1; i + mid - 1 <= n; i++)//起点
    {
        int need = 0;
        for(int j = 1; j <= m; j++)
            need += askRMQ(j, i, i + mid - 1);
        if(need <= K)
        {
            ans[mid] = i;
            return true;
        }
    }
    return false;
}

int main()
{
    memset(ans, 0, sizeof(ans));
    scanf("%d %d %d", &n, &m, &K);
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            scanf("%d", &a[j][i]);
    initRMQ();
    int L = 0, R = n, mid, len = 0;
    while(L <= R)
    {
        mid = (L + R) >> 1;
        if(check(mid))
        {
            len = mid;
            L = mid + 1;
        }
        else
            R = mid - 1;
    }
    for(int i = 1; i <= m; i++)
        printf("%d ", askRMQ(i, ans[len], ans[len] + len - 1));
    return 0;
}