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
",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
",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
",ans);
	}
	return 0;
}