【CERC2015】【BZOJ4432】Greenhouse Growth

Description

你从计算机科学转向农业，你的新工作包括在一个地下温室种植向日葵。在温室中有n个排列在一条直线的向日葵植株，从左向右编号为1到n。有A、B2个照射器为向日葵的生长提供光和热，且照射器A、B分别放置在向日葵的左右两端。
每天只有1个照射器被打开，使所有向日葵转向光源，并使部分向日葵生长。向日葵会生长当且仅当其朝向的相邻植株比它更高，其每天的生长高度为1厘米。请注意，一个植株的生长将使其背后的植株立刻开始生长。
样例数据前三天的生长情况

你将被给出向日葵的初始高度和接下来m天的光照计划，请计算所有向日葵最终的高度。
Input

第一行有2个整n和m(1<= n, m <=300 000)——植株数和天数。
接下来一行包括n个整数h1, h2,… , hn (1 <= hk <=109)——从左到右向日葵的初始高度。
接下来一行包括一个仅含字母A/B长度为m的字符串——从第一天开始的光照计划。
Output

n个整数——从左到右每株向日葵最终的高度
Sample Input

6 5

4 3 5 3 6 6

BABAA

Sample Output

5 5 6 6 6 6
HINT

Source

一开始觉得是线段树,实际上链表模拟一下就行了..
搞标记记录一下A亮的时候长不长,B亮的时候长不长,一个位置长不长,长多少都是可以用当前时间和之前时间来计算的
1表示A亮时候长,2表示B亮时候长,3表示两种都长

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define MAXN 300010
#define GET (ch>='0'&&ch<='9')
#define Abs(x) (x>=0?x:-x)
using namespace std;
inline void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int Tim,n,m,top,tp;
int cnt[MAXN][4];
struct edge;
struct seg;
vector<edge*> now[MAXN][4];
struct edge
{
    seg *pre,*nxt;
    inline void solve();
    inline void merge();
}List[MAXN<<1];
struct seg
{
    int l,r,flag,h0,t0;
    edge *pre,*nxt;
    inline int calc()   {   return h0+cnt[Tim][flag]-cnt[t0][flag]; }
    inline bool checka()    {   return pre&&pre->pre->calc()>calc();   }
    inline bool checkb()    {   return nxt&&nxt->nxt->calc()>calc();   }
    inline void update()    {   flag=checka()|(checkb()<<1);  }
}*first,*last,list[MAXN<<1];
inline void edge::solve()
{
    int delta=abs(pre->calc()-nxt->calc()),flag=pre->flag^nxt->flag;
    if (delta&&flag&&cnt[Tim][flag]+delta<=300000)   now[cnt[Tim][flag]+delta][flag].push_back(this);
}
inline void edge::merge()
{
    if (!(pre&&nxt))    return;
    int h=pre->calc();
    if (nxt->calc()!=h)  return;
    seg* ret=pre;
    ret->l=pre->l;ret->r=nxt->r;ret->h0=h;ret->t0=Tim;
    ret->pre=pre->pre;ret->nxt=nxt->nxt;ret->update();
    if (ret->pre)    ret->pre->nxt=ret;
    if (ret->nxt)    ret->nxt->pre=ret;
    if (ret->pre)    ret->pre->solve();
    if (ret->nxt)    ret->nxt->solve();
    pre=NULL;nxt=NULL;
}
int main()
{
    in(n);in(m);int h;char ch=' ';
    for (int i=1;i<=n;i++)
    {
        in(h);
        if (last&&last->h0==h)   {   last->r=i;continue;  }
        ++top;list[top].l=i;list[top].r=i;list[top].h0=h;list[top].t0=0;
        if (last)
            ++tp,List[tp].pre=last,List[tp].nxt=&list[top],
            last->nxt=&List[tp],list[top].pre=&List[tp],last->update();
        else    first=&list[top];
        last=&list[top];
    }
    last->update();
    for (seg *i=first;i->nxt;i=i->nxt->nxt)    i->nxt->solve();
    while (ch!='A'&&ch!='B')    ch=getchar();
    for (Tim=1;Tim<=m;Tim++)
    {
        cnt[Tim][1]=cnt[Tim-1][1]+(ch=='A');cnt[Tim][2]=cnt[Tim-1][2]+(ch=='B');
        cnt[Tim][3]=cnt[Tim-1][3]+1;
        for (int f=1;f<=3;f++)
        {
            int it=now[cnt[Tim][f]][f].size();
            for (int i=0;i<it;i++)   now[cnt[Tim][f]][f][i]->merge();
        }
        if (Tim==m)
        {
            for (seg* i=first;i;i=i->nxt?i->nxt->nxt:NULL)
                for (int j=i->l;j<=i->r;++j)
                {
                    if (j>1) putchar(' ');
                    printf("%d",i->calc());
                }
            putchar('\n');
        }
        ch=getchar();
    }
}