Codeforces C. Pluses and Minuses (思维 / 模拟) (Round 90 Rated for Div.2)

传送门

题意： 实现以下伪代码

分析以上伪代码得两种情况：

• 当cur < 0时会break当前for循环，然后cur ++，重新开始
• 当cur始终 > 0时会跑完内部循环，最后终止程序

思路：

• 分析易得若最开始cur始终 > 0则万事大吉，但若过程中遇见n次cur < 0就需要重新循环n几次，显然时间复杂度就高了。
• 我们可以用数组a来记录字符串s的前缀和，并用sum数组记录每次cur < 0的位置，用tmp记录上一次cur < 0时的前一位置的前缀和并不断更新
• cur会在每个sum[i]位置重新开始，而该位置的前缀和为b[i]，然后每次res可直接加上每次重新开始的位置的前缀和*经过该位置的次数

代码实现：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 1e6 + 5;

int t, a[N], b[N], sum[N];
char s[N];

signed main()
{
    IOS;

    cin >> t;
    while(t --){
        cin >> s + 1;
        int len = strlen(s + 1);
        int res = 0, tt = 0, tmp = -1;
        a[0] = b[1] = sum[1] = 0;
        for(int i = 1; i <= len; i ++){
            if(s[i] == '+') a[i] = a[i - 1] + 1;
            else a[i] = a[i - 1] - 1;
            if(a[i] <= tmp){
                b[++ tt] = a[i];
                sum[tt] = i;
                tmp = a[i] - 1;
            }
        }
        //res 加上第一次和最后一次循环的次数
        res += (-1 * b[1]) * sum[1] + len;
        for(int i = 2; i <= tt; i ++){
            //负数是小数减大数才为正
            int cur = b[i - 1] - b[i];
            res += cur * sum[i];
        }
        cout << res << endl;
    }

    return 0;
}