CF1202C You Are Given a WASD-string...
作者:互联网
调试题!
*2100
考虑答案由 \(\max(x),\min(x),\max(y),\min(y)\) 决定。
于是枚举一个位置,就拿 \(x\) 举例吧。
那么就要考虑 \([1,pos]\) 的累加最值,枚举当前 \(+-1\),到当前枚举的位置的值,后缀的累加最值加上当前枚举的贡献,三者取个最值,那么就得到了。
有点难写和难调,注意 \((0,0)\) 也得算进去。
#include <bits/stdc++.h>
#define int long long
#define pb push_back
using namespace std;
const int N=(int)(2e5+5),inf=(int)(2e18);
char s[N];
int n,ans,mx[N],mi[N];
void solve() {
cin>>s+1; n=strlen(s+1);
ans=inf;
int x=0,y=0,mxx=0,mix=0,mxy=0,miy=0;
for(int i=1;i<=n;i++) {
if(s[i]=='W') ++y;
if(s[i]=='A') --x;
if(s[i]=='S') --y;
if(s[i]=='D') ++x;
mxx=max(mxx,x); mix=min(mix,x);
mxy=max(mxy,y); miy=min(miy,y);
}
ans=(mxx-mix+1)*(mxy-miy+1);
// cout<<ans<<'\n';
x=0;
for(int i=0;i<=n+1;i++) mi[i]=inf,mx[i]=-inf;
for(int i=1;i<=n;i++) {
if(s[i]=='A') --x,mx[i]=x,mi[i]=x;
if(s[i]=='D') ++x,mx[i]=x,mi[i]=x;
}
int qwq=max(mxx,x+1),qaq=min(mix,x+1);
ans=min(ans,(qwq-qaq+1)*(mxy-miy+1));
qwq=max(mxx,x-1); qaq=min(mix,x-1);
ans=min(ans,(qwq-qaq+1)*(mxy-miy+1));
int premx=0,premi=0;
// cout<<ans<<'\n';
for(int i=n-1;i>=1;i--) mi[i]=min(mi[i+1],mi[i]),mx[i]=max(mx[i+1],mx[i]);
int res=0;
for(int i=0;i<n;i++) {
if(i) {
if(s[i]=='A') --res;
if(s[i]=='D') ++res;
}
int nmxx=max(premx,max(res-1,mx[i+1]-1)),nmix=min(premi,min(res-1,mi[i+1]-1));
if(mx[i+1]!=-inf&&nmxx>=nmix) ans=min(ans,(nmxx-nmix+1)*(mxy-miy+1));
nmxx=max(premx,max(res+1,mx[i+1]+1)); nmix=min(premi,min(res+1,mi[i+1]+1));
if(mx[i+1]!=-inf&&nmxx>=nmix) ans=min(ans,(nmxx-nmix+1)*(mxy-miy+1));
premx=max(premx,res); premi=min(premi,res);
}
// cout<<ans<<'\n';
y=0;
for(int i=0;i<=n+1;i++) mi[i]=inf,mx[i]=-inf;
for(int i=1;i<=n;i++) {
if(s[i]=='W') ++y,mx[i]=y,mi[i]=y;
if(s[i]=='S') --y,mx[i]=y,mi[i]=y;
}
qwq=max(mxy,y+1),qaq=min(miy,y+1);
ans=min(ans,(qwq-qaq+1)*(mxx-mix+1));
qwq=max(mxy,y-1); qaq=min(miy,y-1);
ans=min(ans,(qwq-qaq+1)*(mxx-mix+1));
for(int i=n-1;i>=1;i--) mi[i]=min(mi[i+1],mi[i]),mx[i]=max(mx[i+1],mx[i]);
// for(int i=1;i<=n;i++) cout<<mi[i]<<" - "<<mx[i]<<'\n';
res=0; premx=0,premi=0;
for(int i=0;i<n;i++) {
if(i) {
if(s[i]=='W') ++res;
if(s[i]=='S') --res;
}
// cout<<i<<" "<<res<<'\n';
int nmx=max(premx,max(res-1,mx[i+1]-1)),nmi=min(premi,min(res-1,mi[i+1]-1));
if(mx[i+1]!=-inf&&nmx>=nmi) ans=min(ans,(nmx-nmi+1)*(mxx-mix+1));
nmx=max(premx,max(res+1,mx[i+1]+1)); nmi=min(premi,min(res+1,mi[i+1]+1));
// cout<<nmx<<" : "<<nmi<<'\n';
if(mx[i+1]!=-inf&&nmx>=nmi) ans=min(ans,(nmx-nmi+1)*(mxx-mix+1));
premx=max(premx,res); premi=min(premi,res);
}
// cout<<ans<<'\n';
cout<<ans<<'\n';
}
signed main() {
cin.tie(0); ios::sync_with_stdio(false);
int T; cin>>T; while(T--) solve();
return 0;
}
标签:...,Given,res,min,int,max,ans,CF1202C,mx 来源: https://www.cnblogs.com/xugangfan/p/16479899.html