2534 - Arrow Flips

flip.cpp

//http://coj.uci.cu/24h/problem.xhtml?abb=2534
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>

using namespace std;

int main() {

	int T, K;
	int inicio, fin, k, maxima, window;
	char S[100005];

	scanf("%d", &T);

while(T--) {
	scanf("%d %s", &K, S);
	maxima = 0;
	window = 0; inicio = 0; fin = 0; k = 0;
	while(fin < (int)strlen(S)) {
		if(S[fin] == '>') {
			window++; fin++;
			if(window > maxima) maxima = window;
		}
		else {
			if(k < K) {
				k++; window++; fin++;
				if(window > maxima) maxima = window;
			} else {
				if(S[inicio] == '<') { 
					k--;
					window--;
					inicio++;
				} else {
					window--;
					inicio++;
				}
			}
		}
	}
	window = 0; inicio = 0; fin = 0; k = 0;
	while(fin < (int)strlen(S)) {
		if(S[fin] == '<') {
			window++; fin++;
			if(window > maxima) maxima = window;
		}
		else {
			if(k < K) {
				k++; window++; fin++;
				if(window > maxima) maxima = window;
			} else {
				if(S[inicio] == '>') {
					k--;
					window--;
					inicio++;
				} else {
					window--;
					inicio++;
				}
			}
		}
	}
	printf("%d\n", maxima);
}
	return 0;
}