二分 51Nod1686 第K大区间

传送门：点击打开链接

题意：定义一个区间的值为其众数出现的次数。
现给出n个数，求将所有区间的值排序后，第K大的值为多少。

二分第k大的数是多少，设为m，之后的check，只需要计算有多少个区间的众数的次数是>=m的即可

这里是枚举区间右端点r，然后通过链表来快速找到最靠近右边的左端点l，使得[l,r]中有数字出现了m次，check的复杂度只有O(n)

所以总的复杂度O(nlogn)

#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <cstdio>
#include <cctype>
#include <bitset>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
#define fuck(x) cout<<"["<<x<<"]";
#define FIN freopen("input.txt","r",stdin);
#define FOUT freopen("output.txt","w+",stdout);
//#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int MX = 1e2 + 5;
const int MS = 1e4 + 5;
const int INF = 0x3f3f3f3f;

int n, m, k;
char W[MX][MX];
int A[MS], MAX[MS][15], vis[MX][MX];
int dx[15], dy[15], dq[MS];
void RMQ_init(int n) {
    for(int i = 0; i < n + 1; i++) {
        MAX[i][0] = A[i];
    }
    for(int j = 1; (1 << j) <= n + 1; j++) {
        for(int i = 0; i + (1 << j) - 1 < n + 1; i++) {
            MAX[i][j] = max(MAX[i][j - 1], MAX[i + (1 << (j - 1))][j - 1]);
        }
    }
}
int RMQ_max(int L, int R) {
    int k = 0;
    while((1 << (k + 1)) <= R - L + 1) k++;
    return max(MAX[L][k], MAX[R - (1 << k) + 1][k]);
}
bool umax(int &a, int b) {
    if(b > a) {
        a = b; return true;
    }
    return false;
}
inline int ID(int x, int y) {
    return (x - 1) * m + y;
}
int solve() {
    int ret = 0;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= m; j++) {
            int Max = 0, id = ID(i, j), dsz = 0;
            for(int q = 1; q <= k; q++) {
                int nx = i + dx[q], ny = j + dy[q];
                if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
                vis[nx][ny] = id;

                for(int w = 1; w <= k; w++) {
                    int nnx = nx - dx[w], nny = ny - dy[w];
                    if(nnx < 1 || nnx > n || nny < 1 || nny > m) continue;
                    dq[++dsz] = ID(nnx, nny);
                }
            }

            dq[++dsz] = 0;
            dq[++dsz] = m * n + 1;
            sort(dq + 1, dq + 1 + dsz);
            dsz = unique(dq + 1, dq + 1 + dsz) - dq - 1;

            for(int q = 1; q <= dsz - 1; q++) {
                if(dq[q] + 1 <= dq[q + 1] - 1) umax(Max, RMQ_max(dq[q] + 1, dq[q + 1] - 1));
            }
            for(int q = 2; q <= dsz - 1; q++) {
                int x = (dq[q] - 1) / m + 1, y = (dq[q] - 1) % m + 1, sum = A[ID(x, y)];
                for(int w = 1; w <= k; w++) {
                    int nx = x + dx[w], ny = y + dy[w];
                    if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
                    if(vis[nx][ny] == id) sum -= W[nx][ny] - '0';
                }
                umax(Max, sum);
            }
            umax(ret, Max + A[id]);
        }
    }
    return ret;
}
int main() {
    //FIN;
    int T; scanf("%d", &T);
    while(T--) {
        scanf("%d%d%d", &n, &m, &k);
        for(int i = 1; i <= n; i++) {
            for(int j = 1; j <= m; j++) {
                vis[i][j] = A[ID(i, j)] = 0;
            }
        }

        for(int i = 1; i <= n; i++) scanf("%s", W[i] + 1);
        for(int i = 1; i <= k; i++) {
            scanf("%d%d", &dx[i], &dy[i]);
        }

        for(int i = 1; i <= n; i++) {
            for(int j = 1; j <= m; j++) {
                int id = ID(i, j);
                for(int q = 1; q <= k; q++) {
                    int nx = i + dx[q], ny = j + dy[q];
                    if(nx < 1 || nx > n || ny < 1 || ny > m) continue;
                    A[id] += W[nx][ny] - '0';
                }
            }
        }
        RMQ_init(n * m);
        printf("%d\n", solve());
    }
    return 0;
}