C. Pluses and Minuses-----------------------思維

在這裏插入圖片描述
在這裏插入圖片描述
解析:
求貢獻。
用一個前綴和來表示當前i是否重新cur++
假設前綴和爲sum
當前第i步,sum<0 那麼對答案的貢獻就是i 。然後對應僞代碼,我們要給cur=i+1,
當前第i步,sum>=0 直接跳過

最終還需加上整個長度,因爲如果前綴和都不爲0的話,我們就相當於全部遍歷一遍。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+1000;
typedef long long ll;
char s[N];
int t;
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>(s+1);
		int n=strlen(s+1);
		ll sum=0;
		ll cur=0;
		ll mi=0x3f3f3f3f;
		int f=0;
		for(int i=1;i<=n;i++)
		{
			if(s[i]=='-') cur--;
			else cur++;
			if(cur<0)
			{
				f=1;
				sum+=i;
				cur+=1;
			}
		}
		sum+=n;
		if(f==0) cout<<n<<endl;
		else 
		cout<<sum<<endl;
	
	}
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章