#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using db = long double; // or double, if TL is tight
using str = string; // yay python!
// pairs
using pii = pair<int, int> ;
using pl = pair<ll, ll> ;
using pd = pair<db, db> ;
#define mp make_pair
#define f first
#define s second
#define tcT template < class T
#define tcTU tcT, class U
// ^ lol this makes everything look weird but I'll try it
tcT > using V = vector<T> ;
tcT, size_t SZ > using AR = array<T, SZ> ;
using vi = V<int> ;
using vb = V<bool> ;
using vl = V<ll> ;
using vd = V<db> ;
using vs = V<str> ;
using vpi = V<pii> ;
using vpl = V<pl> ;
using vpd = V<pd> ;
// vectors
// oops size(x), rbegin(x), rend(x) need C++17
#define sz(x) int((x).size())
#define bg(x) begin(x)
#define all(x) bg(x), end(x)
#define rall(x) x.rbegin(), x.rend()
#define sor(x) sort(all(x))
#define rsz resize
#define ins insert
#define pb push_back
#define eb emplace_back
#define ft front()
#define bk back()
#define lb lower_bound
#define ub upper_bound
#define FOR(i, a, b) for (int i = (a); i < (b); ++i)
#define F0R(i, a) FOR(i, 0, a)
#define ROF(i, a, b) for (int i = (b) - 1; i >= (a); --i)
#define R0F(i, a) ROF(i, 0, a)
#define rep(a) F0R(_, a)
#define each(a, x) for (auto &a: x)
const int MOD = 998244353;
const int MX = 2e5 + 5;
const ll BIG = 1e18; // not too close to LLONG_MAX
const db PI = acos((db) - 1);
const int dx[4]
{
1, 0, -1, 0
}, dy[4]
{
0, 1, 0, -1
}; // for every grid problem!!
mt19937 rng((uint32_t) chrono::steady_clock::now().time_since_epoch().count());
template < class T > using pqg = priority_queue<T, vector < T>, greater < T>> ;
struct DSU
{
vi e;
void init(int N)
{
e = vi(N, -1);
}
int get(int x)
{
return e[x] < 0 ? x : e[x] = get(e[x]);
}
bool sameSet(int a, int b)
{
return get(a) == get(b);
}
int size(int x)
{
return -e[get(x)];
}
bool unite(int x, int y)
{
// union by size
x = get(x), y = get(y);
if (x == y) return 0;
if (e[x] > e[y]) swap(x, y);
e[x] += e[y];
e[y] = x;
return 1;
}
};
/*
inline namespace Helpers {
//////////// is_iterable
// https://stackoverflow.com/questions/13830158/check-if-a-variable-type-is-iterable
// this gets used only when we can call begin() and end() on that type
tcT, class = void > struct is_iterable : false_type {};
tcT > struct is_iterable<T, void_t<decltype(begin(declval < T>())),
decltype(end(declval < T>()))
>
> : true_type {};
tcT > constexpr bool is_iterable_v = is_iterable<T>::value;
//////////// is_readable
tcT, class = void > struct is_readable : false_type {};
tcT > struct is_readable<T,
typename std::enable_if_t<
is_same_v < decltype(cin >> declval<T&>()), istream&>
>
> : true_type {};
tcT > constexpr bool is_readable_v = is_readable<T>::value;
//////////// is_printable
// // https://nafe.es/posts/2020-02-29-is-printable/
tcT, class = void > struct is_printable : false_type {};
tcT > struct is_printable<T,
typename std::enable_if_t<
is_same_v < decltype(cout <<declval < T>()), ostream&>
>
> : true_type {};
tcT > constexpr bool is_printable_v = is_printable<T>::value;
}*/
using ll = long long;
using db = long double; // or double, if TL is tight
using str = string; // yay python!
// pairs
using pii = pair<int, int> ;
using pl = pair<ll, ll> ;
using pd = pair<db, db> ;
#define mp make_pair
#define f first
#define s second
#define tcT template < class T
#define tcTU tcT, class U
// ^ lol this makes everything look weird but I'll try it
tcT > using V = vector<T> ;
tcT, size_t SZ > using AR = array<T, SZ> ;
using vi = V<int> ;
using vb = V<bool> ;
using vl = V<ll> ;
using vd = V<db> ;
using vs = V<str> ;
using vpi = V<pii> ;
using vpl = V<pl> ;
using vpd = V<pd> ;
// vectors
// oops size(x), rbegin(x), rend(x) need C++17
#define sz(x) int((x).size())
#define bg(x) begin(x)
#define all(x) bg(x), end(x)
#define rall(x) x.rbegin(), x.rend()
#define sor(x) sort(all(x))
#define rsz resize
#define ins insert
#define pb push_back
#define eb emplace_back
#define ft front()
#define bk back()
#define lb lower_bound
#define ub upper_bound
#define FOR(i, a, b) for (int i = (a); i < (b); ++i)
#define F0R(i, a) FOR(i, 0, a)
#define ROF(i, a, b) for (int i = (b) - 1; i >= (a); --i)
#define R0F(i, a) ROF(i, 0, a)
#define rep(a) F0R(_, a)
#define each(a, x) for (auto &a: x)
template < class T > using pqg = priority_queue<T, vector < T>, greater < T>> ;
/*
inline namespace Helpers {
//////////// is_iterable
// https://stackoverflow.com/questions/13830158/check-if-a-variable-type-is-iterable
// this gets used only when we can call begin() and end() on that type
tcT, class = void > struct is_iterable : false_type {};
tcT > struct is_iterable<T, void_t<decltype(begin(declval < T>())),
decltype(end(declval < T>()))
>
> : true_type {};
tcT > constexpr bool is_iterable_v = is_iterable<T>::value;
//////////// is_readable
tcT, class = void > struct is_readable : false_type {};
tcT > struct is_readable<T,
typename std::enable_if_t<
is_same_v < decltype(cin >> declval<T&>()), istream&>
>
> : true_type {};
tcT > constexpr bool is_readable_v = is_readable<T>::value;
//////////// is_printable
// // https://nafe.es/posts/2020-02-29-is-printable/
tcT, class = void > struct is_printable : false_type {};
tcT > struct is_printable<T,
typename std::enable_if_t<
is_same_v < decltype(cout <<declval < T>()), ostream&>
>
> : true_type {};
tcT > constexpr bool is_printable_v = is_printable<T>::value;
}*/
const int N = 1e6 + 5;
int t, n, a[N], tree[N];
long long pr[N], ans;
void update(long long idx, int val)
{
idx++;
for (int i = idx; i < N; i += i &(-i))
{
tree[i] += val;
}
}
int getans(int idx)
{
idx++;
int pas = 0;
for (int i = idx; i > 0; i -= i &(-i))
{
pas += tree[i];
}
return pas;
}
main()
{
ios_base::sync_with_stdio(false),
cin.tie(NULL), cout.tie(NULL);
cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
int x;
cin >> x;
vector<pair < long long, int>> v;
for (int i = 1; i <= n; i++)
{
a[i] -= x;
pr[i] = pr[i - 1] + a[i];
v.pb({ pr[i], i });
}
v.pb({ 0, 0 });
sort(v.begin(), v.end());
for (int i = 0; i < v.size(); i++)
{
ans += getans(v[i].s);
update(v[i].s, 1);
}
cout << ans << "\n";
}
Compilation message
vudu.cpp:232:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
232 | main()
| ^~~~
vudu.cpp: In function 'int main()':
vudu.cpp:254:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
254 | for (int i = 0; i < v.size(); i++)
| ~~^~~~~~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
2 ms |
724 KB |
Output is correct |
| 2 |
Correct |
1 ms |
596 KB |
Output is correct |
| 3 |
Correct |
1 ms |
596 KB |
Output is correct |
| 4 |
Correct |
220 ms |
30640 KB |
Output is correct |
| 5 |
Correct |
117 ms |
23112 KB |
Output is correct |
| 6 |
Correct |
184 ms |
27288 KB |
Output is correct |
| 7 |
Correct |
236 ms |
28316 KB |
Output is correct |
| 8 |
Correct |
166 ms |
24596 KB |
Output is correct |
| 9 |
Correct |
225 ms |
31696 KB |
Output is correct |
| 10 |
Correct |
200 ms |
27560 KB |
Output is correct |
