【codeforces 733E】Sleep in Class

【题目链接】:http://codeforces.com/problemset/problem/733/E

【题意】

有n级台阶,每个台阶上都有一个tag;
标记着向上或向下;
你到了某级台阶,就要按照tag上的方向走;
然后那个tag所标识的方向会反转;
(可能会从1和n的位置掉出这个台阶);
让你求出初始位置为1..n时,多久会掉出台阶;

【题解】

/*
    以下分析来自http://blog.csdn.net/v5zsq/article/details/71171709
    -----
        假如从一个U开始走，那么就一直往上一直走到一个D，然
        后开始往下走一直到一个U，然后一直往上回到原位，这一
        段走过后，这个点下面第一个U变成了D，上面第一个D变成
        了U，而其他位置都没有变化，走过的步数是这两个位置差
        的二倍，所以每进行一次这样的操作，该点上面一个D和下
        面一个U就会反向，最后一定是某一边全部是D该点从下面
        出去或全部是U然后该点从上面出去，令lu[i]为1~i中U的
        个数，rd[i]表示i~n中D的个数，如果lu[i]<=rd[i+1]，那
        么第i个点出发就会从下面出去，答案为i+i右边前lu[i]个
        D到i的距离和*2+i左边所有U到i的距离和*2，否则第i个点
        就会从上面出去，答案为n-i+1+i左边前rd[i]个U到i的距
        离和*2+i右边所有D到i的距离和*2
    -----
    则我们找到
        最小的i;
        使得lu[i]>rd[i+1];
        (lu[i]是单增的,rd[i]则是单减的)
        (i再变大的话,lu[i]只可能更大,而rd[i+1]只可能更小)
        则1..i-1内;
            每一个位置,左边的u的数目都是小于右边的d的个数的;
            也就是说每个左u都能有一个右d和其对应;
        i..n内
            每一个位置,左边的u的数目都是大于右边的d的个数的;
            也就是说每一个右d都能有一个左u个和它对应;
        记录下每个d和u的位置;
        然后写一个类似栈的东西来维护所有的d到左端点的距离;
        所有的u到左端点的距离;然后前者减后者,就是每个位置两边
        的所有左u和右d的距离和了;
        分成两部分求解;
*/

【Number Of WA】

2

【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define ms(x,y) memset(x,y,sizeof x)

typedef pair<int,int> pii;
typedef pair<LL,LL> pll;

const int dx[9] = {0,1,-1,0,0,-1,-1,1,1};
const int dy[9] = {0,0,0,-1,1,-1,1,-1,1};
const double pi = acos(-1.0);
const int N = 1e6+100;

int n,lu[N],rd[N],cut;
char s[N];
vector <int> u,d;
vector <LL> ans1,ans2;

int main()
{
    //freopen("F:\\rush.txt","r",stdin);
    ios::sync_with_stdio(false),cin.tie(0);//scanf,puts,printf not use
    //init??????
    cin >> n;
    cin >> (s+1);
    rep1(i,1,n)
        lu[i]=lu[i-1]+(s[i]=='U');
    rep2(i,n,1)
        rd[i]=rd[i+1]+(s[i]=='D');
    rep1(i,1,n)
        if (s[i]=='U')
            u.pb(i);
        else
            d.pb(i);
    cut = n+1;
    rep1(i,1,n)
        if (lu[i]>rd[i+1])
        {
            cut = i;
            break;
        }
    LL sum1 = 0,sum2 = 0;
    int pd = 0;
    rep1(i,1,cut-1)
    {
        if (s[i]=='U')
            sum1+=i*2;
        else
            //s[i]=='D'
            sum2-=i*2;
        sum2+=d[pd++]*2;
        ans1.pb(sum2-sum1+i);
    }
    sum1 = 0 ,sum2 = 0;
    int pu = u.size()-1;
    rep2(i,n,cut)
    {
        if (s[i]=='D')
            sum2+=i*2;
        else
            //s[i]=='U'
            sum1-=i*2;
        sum1+=u[pu--]*2;
        ans2.pb(sum2-sum1+n-i+1);
    }
    int len = ans1.size();
    rep1(i,0,len-1)
        cout <<ans1[i]<<' ';
    len = ans2.size();
    rep2(i,len-1,0)
        cout<<ans2[i]<<' ';
    cout << endl;
    return 0;
}