これはどこかで解いていそう。
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; }
まとめ
典型そうな気はする。