2018-09-30 23:01:58 +01:00
---
id: 5987fd532b954e0f21b5d3f6
2020-11-27 19:02:05 +01:00
title: Equilibrium index
2018-09-30 23:01:58 +01:00
challengeType: 5
2019-08-05 09:17:33 -07:00
forumTopicId: 302255
2021-01-13 03:31:00 +01:00
dashedName: equilibrium-index
2018-09-30 23:01:58 +01:00
---
2020-11-27 19:02:05 +01:00
# --description--
2019-03-02 17:42:56 +09:00
An equilibrium index of a sequence is an index into the sequence such that the sum of elements at lower indices is equal to the sum of elements at higher indices.
2020-11-27 19:02:05 +01:00
For example, in a sequence $A$:
< ul style = 'list-style: none;' >
2019-03-02 17:42:56 +09:00
< li > < big > $A_0 = -7$< / big > < / li >
< li > < big > $A_1 = 1$< / big > < / li >
< li > < big > $A_2 = 5$< / big > < / li >
< li > < big > $A_3 = 2$< / big > < / li >
< li > < big > $A_4 = -4$< / big > < / li >
< li > < big > $A_5 = 3$< / big > < / li >
< li > < big > $A_6 = 0$< / big > < / li >
< / ul >
2020-11-27 19:02:05 +01:00
`3` is an equilibrium index, because:
< ul style = 'list-style: none;' >
2019-03-02 17:42:56 +09:00
< li > < big > $A_0 + A_1 + A_2 = A_4 + A_5 + A_6$< / big > < / li >
< / ul >
2020-11-27 19:02:05 +01:00
`6` is also an equilibrium index, because:
< ul style = 'list-style: none;' >
2019-03-02 17:42:56 +09:00
< li > < big > $A_0 + A_1 + A_2 + A_3 + A_4 + A_5 = 0$< / big > < / li >
< / ul >
2020-11-27 19:02:05 +01:00
2019-03-02 17:42:56 +09:00
(sum of zero elements is zero)
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
`7` is not an equilibrium index, because it is not a valid index of sequence $A$.
# --instructions--
2019-03-02 17:42:56 +09:00
Write a function that, given a sequence, returns its equilibrium indices (if any).
2020-11-27 19:02:05 +01:00
2019-03-02 17:42:56 +09:00
Assume that the sequence may be very long.
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
# --hints--
`equilibrium` should be a function.
```js
assert(typeof equilibrium === 'function');
2018-09-30 23:01:58 +01:00
```
2020-11-27 19:02:05 +01:00
`equilibrium([-7, 1, 5, 2, -4, 3, 0])` should return `[3,6]` .
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
```js
assert.deepEqual(equilibrium(equilibriumTests[0]), ans[0]);
```
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
`equilibrium([2, 4, 6])` should return `[]` .
2018-09-30 23:01:58 +01:00
```js
2020-11-27 19:02:05 +01:00
assert.deepEqual(equilibrium(equilibriumTests[1]), ans[1]);
```
2020-09-15 09:57:40 -07:00
2020-11-27 19:02:05 +01:00
`equilibrium([2, 9, 2])` should return `[1]` .
```js
assert.deepEqual(equilibrium(equilibriumTests[2]), ans[2]);
```
`equilibrium([1, -1, 1, -1, 1, -1, 1])` should return `[0,1,2,3,4,5,6]` .
```js
assert.deepEqual(equilibrium(equilibriumTests[3]), ans[3]);
```
`equilibrium([1])` should return `[0]` .
```js
assert.deepEqual(equilibrium(equilibriumTests[4]), ans[4]);
2018-09-30 23:01:58 +01:00
```
2020-11-27 19:02:05 +01:00
`equilibrium([])` should return `[]` .
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
```js
assert.deepEqual(equilibrium(equilibriumTests[5]), ans[5]);
```
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
# --seed--
## --after-user-code--
2018-09-30 23:01:58 +01:00
```js
2018-10-20 21:02:47 +03:00
const equilibriumTests =
[[-7, 1, 5, 2, -4, 3, 0], // 3, 6
[2, 4, 6], // empty
[2, 9, 2], // 1
[1, -1, 1, -1, 1, -1, 1], // 0,1,2,3,4,5,6
[1], // 0
[] // empty
];
const ans = [[3, 6], [], [1], [0, 1, 2, 3, 4, 5, 6], [0], []];
2018-09-30 23:01:58 +01:00
```
2020-11-27 19:02:05 +01:00
## --seed-contents--
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
```js
function equilibrium(a) {
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
}
```
2018-09-30 23:01:58 +01:00
2020-11-27 19:02:05 +01:00
# --solutions--
2018-09-30 23:01:58 +01:00
```js
function equilibrium(a) {
let N = a.length,
i,
l = [],
r = [],
e = [];
for (l[0] = a[0], r[N - 1] = a[N - 1], i = 1; i < N ; i + + )
{ l[i] = l[i - 1] + a[i], r[N - i - 1] = r[N - i] + a[N - i - 1]; }
for (i = 0; i < N ; i + + )
{ if (l[i] === r[i]) e.push(i); }
return e;
}
```