牛客 小苯的Z串匹配

 

注意数组元素是0的情况

#include<iostream>
using namespace std;

int t;
const int N=2e5+10;

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	cin>>t;
	while(t)
	{
		long long n;
		cin>>n;
		long long a[N];
		for(int i=0; i<n; ++i) cin>>a[i];
		string s; cin>>s;	
		
		long long ans=0;
		for(int i=0; i<n; ++i)
		{
			if(s[i]=='<' && a[i]>0)
			{
				ans++;
				a[i]=-2;
			}
			if(s[i]=='<' && a[i]==0)
			{
				ans++;
				a[i]=-2;
			}
			if(s[i]=='>' && a[i]<0)
			{
				ans++;
				a[i]=2;
			}
			if(s[i]=='>' && a[i]==0)
			{
				ans++;
				a[i]=2;
			}
						
			if(s[i]=='Z' && a[i-1]*a[i]<0)
			{
				ans++;
				{
					if(a[i-1]<0) a[i]=-2;
					if(a[i-1]>0) a[i]=2;
				}
			}
			if(s[i]=='Z' && a[i-1]*a[i]==0)
			{
				ans++;
				{
					if(a[i-1]<0) a[i]=-2;
					if(a[i-1]>0) a[i]=2;
				}
			}
			if(s[i]=='Z' && a[i-1]*a[i]==0)
			{
				ans++;
				{
					if(a[i-1]<0) a[i]=-2;
					if(a[i-1]>0) a[i]=2;
				}
			}
		}
		t--;
		
		cout<<ans<<endl;	
	} 	
	
	return 0;
}