【Codeforces 733E】 Sleep in Class 【模拟】

最新推荐文章于 2018-10-25 21:50:07 发布

sjtsjt709

最新推荐文章于 2018-10-25 21:50:07 发布

阅读量434

点赞数

分类专栏： ACM

本文链接：https://blog.csdn.net/sjtsjt709/article/details/53185477

版权

ACM 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

题解：

有一个UD序列，每个点按照当前的UD来走，如果走过一次，该UD就会反向

题意：

从哪里出去
比较cntu和cntd(当前i之前u的个数)(当前i之后d的个数)
当前s[i]=='U' 如果cntd>cntu从左边出去 cntd<=cntu从右边出去
当前s[i]=='D' 如果cntu>cntd从右边出去 cntu<=cntd从左边出去

计数问题
如果从左边出去,ans+=i 如果从右边出去,ans+=n-i+1
每一个合理的u或者d 会对步数贡献2*abs(i-p)
如果s[i]=='U' 从右边出去会有 x个合法的d x个合法的u
从左边出去会有 x+1个合法的d x个合法的u
如果s[i]=='D' 从右边出去会有 x个合法的d x+1个合法的u
从左边出去会有 x个合法的d x个合法的u
x=min(cntu,cntd)

直接计算时间复杂度爆炸
考虑线性计算
设前一个答案为pans，当前合法的u和d，之前合法的pu，pd
当前后移一位 pu个u和当前距离变大1 pd个d和当前距离减小1
然后考虑u和d，这个画图就好了

#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define PB push_back
#define MP make_pair
#define ll long long
#define MS(a,b) memset(a,b,sizeof(a))
#define LL (rt<<1)
#define RR (rt<<1|1)
#define lson l,mid,LL
#define rson mid+1,r,RR
#define pii pair<int,int>
#define pll pair<ll,ll>
#define lb(x) (x&(-x))
void In(){freopen("in.in","r",stdin);}
void Out(){freopen("out.out","w",stdout);}
const int N=1e6+10;
const int M=3e5+10;
const int Mbit=1e6+10;
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
char s[N];
int n,cntu,cntd,posd,posu,x,d,pd,u,pu,flag,i,px;
ll ans,pans;
void getd()
{
    posd++;while(posd<=n&&s[posd]!='D')posd++;
    ans+=2*(posd-i);
}
void getu()
{
    ans-=2*(i-posu);
    posu++;while(posu<i&&s[posu]!='U')posu++;
}
int main()
{
    s[0]='X';
    while(~scanf("%d",&n)){
        scanf("%s",s+1);
        ans=pans=cntu=cntd=posd=posu=pd=pu=0;
        for(i=n;i>=1;i--)
            if(s[i]=='D')cntd++;else posu=i;
        for(i=1;i<=n;i++){
            ans=pans;

            if(s[i-1]=='U')cntu++;
            if(s[i]=='D')cntd--;

            if(s[i]=='U'){if(cntd>cntu)  flag=0;else flag=1;}
            else if(s[i]=='D'){if(cntu<=cntd) flag=0;else flag=1;}

            x=min(cntu,cntd);
            if(s[i]=='U'){
                if(flag)d=x,u=x;
                else d=x+1,u=x;
            }
            else{
                if(flag)d=x,u=x+1;
                else d=x,u=x;
            }

            ans+=2*pu-2*pd;

            if(d>pd)getd();
            else if(d==pd);
            else if(d<pd);
            if(s[i]=='D'&&d>=pd)getd();

            if(u>pu)ans+=2;
            else if(u==pu);
            else if(u<pu)getu();
            if(s[i-1]=='U'&&u<=pu)getu(),ans+=2;

            if(flag)printf("%lld ",ans+n-i+1);
            else printf("%lld ",ans+i);
            pans=ans;pd=d;pu=u;
        }
        printf("\n");
    }
    return 0;
}