백준 14246번 k보다 큰 구간 C++

출처 : https://www.acmicpc.net/problem/14246

i~j의 합이 k보다 작거나 같다면 j를 늘려주고

i~j의 합이 k보다 크다면 i~j+a도 k보다 크기 때문에 뒤에 남은 배열의 수만큼 더해주고, i를 늘려주는 방식으로 문제를 해결하였다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>

using namespace std;
using ll = long long;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n, k;
	cin >> n;
	vector<int> v(n);
	for (int i = 0; i < n; i++) {
		cin >> v[i];
	}
	cin >> k;
	int left = 0;
	int right = 0;
	ll cnt = 0;
	ll sum = v[0];
	while (true) {
		if (right == n - 1 && left == n - 1) break;
		if (right == n - 1) {
			if (sum > k) {
				cnt += n - right;
			}
			sum -= v[left];
			left++;	
		}
		else {
			if (sum <= k) {
				right++;
				sum += v[right];
			}
			else {
				cnt += n - right;
				sum -= v[left];
				left++;
			}
		}
	}
	cout << cnt;
};