fbpx

Problema #2322 – Merry – Rezolvari PBInfo

de Mihai-Alexandru

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;
}
Comentarii

S-ar putea sa iti placa