题目大意:
一个n*m的格子上有b个障碍物,现在让你往上面放正方形(长宽在格线上)。问可以放多少种边长、位置不同的正方形?
思路:
很容易想到一个O(n^3)的暴力:
首先前缀和,然后枚举某一个顶点和正方形的边长,判断一下正方形里面是否为空,如果空,则为一种满足条件的答案。
枚举边长可以改成二分,这样复杂度是O(n^2 log n)的。
再考虑一个O(n^2)的动规:
用f[i][j]保存以(i,j)为右下角顶点的正方形的个数,显然,如果(i-1,j)(i,j-1)(i-1,j-1)(i,j)上都没有障碍,那么f[i][j]=min(f[i-1][j],f[i][j-1],f[i-1][j-1])+1。
正解是一个O(nb)的奇怪做法:
首先对于障碍物按列再按行排序。
枚举每一行,将在这一行上面的障碍物加入一个数组中。
考虑下边界在当前行的极大化正方形,无非有以下两种情况:
1.上边被顶到。
2.左右两边被顶到。
现在我们让每个障碍物“代表”被其约束的点,记录下行数比它大的左端点和右端点,这一过程可以用单调栈来求。
最后分情况计算出符合条件的正方形个数即可。
1 #include<stack> 2 #include<vector> 3 #include<cstdio> 4 #include<cctype> 5 #include<algorithm> 6 typedef long long int64; 7 inline int getint() { 8 register char ch; 9 while(!isdigit(ch=getchar())); 10 register int x=ch^'0'; 11 while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0'); 12 return x; 13 } 14 const int B=1001; 15 struct Point { 16 int x,y; 17 bool operator < (const Point &another) const { 18 if(y==another.y) return x>another.x; 19 return y<another.y; 20 } 21 }; 22 std::vector<Point> a,p; 23 std::stack<int> q; 24 int l[B],r[B]; 25 inline int calc(const int &x,const int &y) { 26 if(!x) return 0; 27 return (int64)(y*2-std::min(x,y))*(std::min(x,y)-1)/2; 28 } 29 int main() { 30 int n=getint(),m=getint(),b=getint(); 31 for(register int i=1;i<=b;i++) { 32 const int x=getint(),y=getint(); 33 a.push_back((Point){x,y}); 34 } 35 std::sort(a.begin(),a.end()); 36 int64 ans=0; 37 for(register int i=1;i<=n;i++) { 38 p.clear(); 39 p.push_back((Point){i,0}); 40 for(register unsigned j=0;j<a.size();j++) { 41 if(a[j].x<=i) { 42 p.push_back(a[j]); 43 } 44 } 45 p.push_back((Point){i,m+1}); 46 while(!q.empty()) q.pop(); 47 q.push(0); 48 for(register unsigned i=1;i<p.size();i++) { 49 while(q.size()>1&&p[q.top()].x<=p[i].x) q.pop(); 50 l[i]=q.top(); 51 q.push(i); 52 } 53 while(!q.empty()) q.pop(); 54 q.push(p.size()-1); 55 for(register unsigned i=p.size()-2;i>0;i--) { 56 while(q.size()>1&&p[q.top()].x<p[i].x) q.pop(); 57 r[i]=q.top(); 58 q.push(i); 59 } 60 for(register unsigned j=1;j<p.size();j++) { 61 ans+=calc(i,p[j].y-p[j-1].y-1); 62 } 63 for(register unsigned j=1;j<p.size()-1;j++) { 64 ans+=calc(i-p[j].x,p[r[j]].y-p[l[j]].y-1)-calc(i-p[j].x,p[r[j]].y-p[j].y-1)-calc(i-p[j].x,p[j].y-p[l[j]].y-1); 65 } 66 } 67 printf("%lld ",ans); 68 return 0; 69 }