传送门:点击打开链接
题意:定义一个区间的值为其众数出现的次数。
现给出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;
}