Cerința
Într-un ținut îndepărtat cu peisaje de poveste pacea domnește peste întreg ținutul deoarece lumea are absolut tot ce îi trebuie.
Moș Crăciun însa dorește să le mai aducă zâmbetul pe buze și de aceea a decis să ducă fiecărei case câte o sticlă de șampanie deoarece deși oamenii de aici au bani, nu au magazine de unde să cumpere. Însă din păcate Moș Crăciun este bătrân și nu poate transporta prea multe sticle așa că decide să distribuie câte o sticla de șampanie la fiecare două case. Pentru simplitate Moș Crăciun a făcut o hartă a caselor în care o casă este notată cu 1
și pământul gol cu 0
. Moș Crăciun însă nu prea le are cu programarea și de aceea vă roagă să îl ajutați!
Scrieți un program care primind harta caselor afișează câte sticle de șampanie trebuie să aducă Moș Crăciun.
Date de intrare
Programul va citi un număr natural n
de pe prima linie și de pe a doua linie n
numere naturale din intervalul [0;1]
.
Date de ieșire
Programul va afișa numărul s
de sticle necesare
Restricții și precizări
1 ≤ n ≤ 1.000.000
- cele
n
numere citite vor fi din intervalul[0;1]
- desigur șampania oferită de Moș Crăciun este una fără alcool
- oricare două case sunt prietene și nu există pereche de case care să refuze șampania
Exemplu
Intrare
7 1 0 0 1 0 1 1
Ieșire
6
Explicație
Există 6 perechi de case ce vor primi sticla de șampanie cele cu indici 1
și 4
, 1
și 6
, 1
și 7
, 4
și 6
, 4
și 7
, 6
și 7
.
#include <bits/stdc++.h> using namespace std; int main() { long long n , cnt = 0 , x; cin >> n; for(int i = 1 ; i <= n ;++i) { cin >> x; if(x == 1) cnt ++; } if(cnt == 1) cout << 1; else cout << cnt * (cnt - 1) / 2; }