結果

問題 No.1535 五七五
コンテスト
ユーザー Example0911
提出日時 2021-06-06 18:26:36
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 666 ms / 2,000 ms
コード長 866 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,007 ms
コンパイル使用メモリ 205,392 KB
最終ジャッジ日時 2025-01-22 04:01:32
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include "bits/stdc++.h"
#define int long long
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
const ll INF = (1LL << 61);
ll mod = 998244353;

signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int N; cin >> N;
	int a, b, c; cin >> a >> b >> c;
	vector<int>A(N);
	for (int i = 0; i < N; i++) {
		string S; cin >> S;
		A[i] = (int)S.size();
	}
	vector<int>sum(N + 1);
	for (int i = 0; i < N; i++) {
		sum[i + 1] = sum[i] + A[i];
	}
	map<int, int>mp, mp2;
	for (int i = 0; i <= N; i++) {
		mp2[sum[i] - c]++;
	}
	mp[sum[0] + a]++;
	int ans = 0;
	for (int i = 0; i < N; i++) {
		mp2[sum[i] - c]--;
		int idx = lower_bound(sum.begin(), sum.end(), sum[i + 1] + b) - sum.begin();
		if (idx <= N && sum[idx] - sum[i + 1] == b) {
			ans += mp[sum[i + 1]] * mp2[sum[idx]];
			
		}
		mp[sum[i + 1] + a]++;
	}
	cout << ans << endl;

	return 0;

}
0