kmjp's blog

競技プログラミング参加記です

yukicoder : No.2574 Defect-free Rectangles

これはどこかで解いていそう。
https://yukicoder.me/problems/no/2574

問題

H*Wのグリッドがあり、一部穴の開いたマスがある。
このグリッドの長方形区間のうち、穴の開いたマスを含まないものは何通りか。

解法

下端を総当たりし、横向きに右下端を平面走査しよう。
スタックを使って、穴の開いてないマスの高さに対し、どこまで左端を伸ばせるか保持しながら、左上端の候補数を数え上げて行く。

int H,W,N;
int U[3030][3030];

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	MINUS(U);
	cin>>H>>W>>N;
	FOR(i,N) {
		cin>>y>>x;
		U[y-1][x-1]=y-1;
	}
	FOR(x,W) {
		for(y=1;y<H;y++) U[y][x]=max(U[y][x],U[y-1][x]);
	}
	ll ret=0;
	FOR(y,H) {
		vector<pair<int,int>> V={{0,-1}};
		int sum=0;
		FOR(x,W) {
			int h=y-U[y][x];
			sum+=h;
			while(V.back().first>h) {
				auto p=V.back();
				V.pop_back();
				if(V.back().first>h) {
					sum-=(p.first-V.back().first)*(x-p.second);
				}
				else {
					sum-=(p.first-h)*(x-p.second);
					V.push_back({h,p.second});
					break;
				}
			}
			V.push_back({h,x});
			ret+=sum;
		}
	}
	cout<<ret<<endl;
	
}

まとめ

典型そうな気はする。