题目链接:BZOJ - 1303
题目分析
首先,找到 b 的位置 Pos, 然后将数列中小于 b 的值赋为 -1 ,大于 b 的值赋为 1 。
从 b 向左扩展,不断算 Sum[i, b - 1] ,然后将 Cnt[Sum[i, b - 1]] 加一,这样就算出每个左边的Sum值有多少个了。
然后从 b 向右扩展,不断算 Sum[b + 1, i] ,然后 Ans += Cnt[Sum[b + 1, i]] 。
代码
#include <iostream> #include <cstdio> #include <cstdlib> #include <cstring> #include <cmath> #include <algorithm> using namespace std; const int MaxN = 100000 + 5; int n, b, num, Pos, Sum; int A[MaxN], CntL[MaxN * 2]; typedef long long LL; LL Ans; int main() { scanf("%d%d", &n, &b); for (int i = 1; i <= n; ++i) { scanf("%d", &num); if (num == b) { A[i] = 0; Pos = i; continue; } if (num < b) A[i] = -1; else A[i] = 1; } Sum = 0; for (int i = Pos; i >= 1; --i) { Sum += A[i]; ++CntL[Sum + n]; } Sum = 0; Ans = 0ll; for (int i = Pos; i <= n; ++i) { Sum += A[i]; Ans += (LL)CntL[-Sum + n]; } printf("%lld\n", Ans); return 0; }
时间: 2024-10-19 15:03:17