codeforces 407D Largest Submatrix 3
codeforces 407D Largest Submatrix 3
题意
找出最大子矩阵,须满足矩阵内的元素互不相等。
题解
官方做法
http://codeforces.com/blog/entry/11333
- \(O(n^6)\)
- 枚举子矩阵,暴力check。
- \(O(n^4)\)
- 枚举上下边界,双指针。
- \(O(n^3log_2n)\)
- 假设当前上边界 \(up\), 下边界 \(down\),\(R_i\) 表示当 \(i\) 为左边界时,右边界最大是 \(R_i\)。
- 当 \(down->down+1\) 时,\(R_i\) 要么不变,要么减小。
- 减小的情况:\(a_{down+1, j}(i<=j<=R_i)\) 在矩阵 \((up, i, down, R_i)\) 中出现。
- 因此对于每个 \(a_{down+1, j}(i<=j<=R_i)\),我们需要寻找在矩阵 \((up, i, down, R_i)\) 中出现的,从左边最靠近 \(j\),和从右边最靠近 \(j\) 的两个位置。这个可以用set维护。
- \(O(n^3)\)
- 如何优化掉 \(O(n^3log_2n)\) 做法的 \(log_2n\) ?
- 从大到小枚举 \(up\)。当 \(up->up-1\) 时,\(O(n^2)\) 更新 \((up, 1, n, m)\) 中所有点的从左边最靠近,和从右边最靠近的两个位置。
区间dp做法
\(f_{i, l, r}\)表示下边界是 \(i\), 左右边界是 \(l, r\) 时,上边界的值。
\(f_{i, l, r}=max\{f_{i-1, l, r}, f_{i, l+1, r}, f_{i, l, r-1}, a_{i, l}与a_{1-i, r}的限制, a_{i, r}与a_{1-i, l}的限制\}\)
可以这样理解:\(f_{i, l, r}\) 需要维护A、B、C、D、E、F这六个块之间的关系,\(f_{i-1, l, r}\) 只维护了A、B、C之间的关系,\(f_{i, l+1, r}\) 只维护了B、C、E、F之间的关系,\(f_{i, l, r-1}\) 只维护了A、B、D、E之间的关系,剩下的还需维护A与F、C与D、D与F的关系。
// 代码实现可以省掉第一维。
代码
\(O(n^3log_2n)\)做法
T了 不知道有没有写错
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define rep(i, a, b) for(int i=(a); i<(b); i++)
#define sz(x) (int)x.size()
#define de(x) cout<< #x<<" = "<<x<<endl
#define dd(x) cout<< #x<<" = "<<x<<" "
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int N=404;
int n,m;
int a[N][N], R[N], cnt[N*N];
set<int> pos[N*N];
int solve() {
int ans=1;
rep(up,1,n+1) {
rep(i,0,N*N) pos[i].clear();
rep(i,0,m+1) R[i]=m;
rep(down,up,n+1) {
int r=0;
rep(i,1,m+1) {
while(r<m&&!cnt[a[down][r+1]]) {
++r;
++cnt[a[down][r]];
}
R[i]=min(R[i], r);
--cnt[a[down][i]];
}
rep(j,1,m+1) {
int c=a[down][j];
auto it1=pos[c].upper_bound(j);
auto it2=pos[c].lower_bound(j);
if(it1!=pos[c].begin()) {
--it1;
if(*it1<=j) R[*it1]=min(R[*it1], j-1);
}
if(it2!=pos[c].end()) {
if(j<=*it2) R[j]=min(R[j], *it2-1);
}
}
for(int i=m-1;i;--i) R[i]=min(R[i], R[i+1]);
rep(i,1,m+1) if(i<=R[i]) ans=max(ans, (down-up+1)*(R[i]-i+1));
rep(i,1,m+1) pos[a[down][i]].insert(i);
}
}
return ans;
}
int main() {
while(~scanf("%d%d",&n,&m)) {
///read
rep(i,1,n+1) rep(j,1,m+1) scanf("%d",&a[i][j]);
///solve
printf("%d\n",solve());
}
return 0;
}
\(O(n^3)\) 做法
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define rep(i, a, b) for(int i=(a); i<(b); i++)
#define sz(x) (int)x.size()
#define de(x) cout<< #x<<" = "<<x<<endl
#define dd(x) cout<< #x<<" = "<<x<<" "
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int N=404;
int n, m;
int a[N][N], p[N*N], l[N][N], r[N][N], R[N];
void Min(int &a, int b) {
if(a>b) a=b;
}
void Max(int &a, int b) {
if(a<b) a=b;
}
int solve() {
int ans=1;
rep(i,0,n+1) rep(j,0,m+1) l[i][j]=0, r[i][j]=m+1;
for(int u=n;u;--u) {
rep(i,1,m+1) p[a[u][i]]=0;
rep(i,1,m+1) l[u][i]=p[a[u][i]], p[a[u][i]]=i;
rep(i,1,m+1) p[a[u][i]]=m+1;
for(int i=m;i;--i) r[u][i]=p[a[u][i]], p[a[u][i]]=i;
memset(p,0,sizeof(p));
rep(i,1,m+1) {
p[a[u][i]]=i;
rep(v,u+1,n+1) Max(l[v][i], p[a[v][i]]);
}
rep(i,0,N*N) p[i]=m+1;
for(int i=m;i;--i) {
p[a[u][i]]=i;
rep(v,u+1,n+1) Min(r[v][i], p[a[v][i]]);
}
rep(i,1,m+1) R[i]=m+1;
rep(v,u,n+1) {
rep(i,1,m+1) {
Min(R[l[v][i]], i);
Min(R[i], r[v][i]);
}
for(int i=m-1;i;--i) Min(R[i], R[i+1]);
rep(i,1,m+1) {
Max(ans, (v-u+1)*(R[i]-i));
}
}
}
return ans;
}
int main() {
while(~scanf("%d%d",&n,&m)) {
///read
rep(i,1,n+1) rep(j,1,m+1) scanf("%d",&a[i][j]);
///solve
printf("%d\n",solve());
}
return 0;
}
区间dp做法
#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define rep(i, a, b) for(int i=(a); i<(b); i++)
#define sz(x) (int)x.size()
#define de(x) cout<< #x<<" = "<<x<<endl
#define dd(x) cout<< #x<<" = "<<x<<" "
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int N=404;
int n,m;
int a[N][N], f[N][N], p[N][160004];
int main() {
while(~scanf("%d%d",&n,&m)) {
rep(i,1,n+1) rep(j,1,m+1) scanf("%d",&a[i][j]);
memset(f,0,sizeof(f));
memset(p,0,sizeof(p));
int ans=1;
rep(i,1,n+1) {
rep(len,1,m+1) {
for(int l=1, r=len;r<=m;++l, ++r) {
if(l==r) {
f[l][r]=max(f[l][r], p[l][a[i][l]]);
} else {
f[l][r]=max(f[l][r], max(f[l][r-1], f[l+1][r]));
f[l][r]=max(f[l][r], max(p[l][a[i][r]], p[r][a[i][l]]));
if(a[i][l]==a[i][r]) f[l][r]=i;
}
ans=max(ans, (i-f[l][r])*(r-l+1));
}
}
rep(j,1,m+1) p[j][a[i][j]]=i;
}
printf("%d\n",ans);
}
return 0;
}