Pairwise Sum and Divide

  • + 0 comments

    Simple question

    #include <cmath>
    #include <cstdio>
    #include <vector>
    #include <iostream>
    #include <algorithm>
    using namespace std;
    
    
    int main() {int t;
                cin>>t;
                while(t--)
                {
                    long int n;
                    cin>>n;
                    long long int a[n];
                    long long int s=0;
                    long int c1=0,c2=0;
                    for(int i=0;i<n;i++)
                    {
                        cin>>a[i];
                        if(a[i]==1)c1++;
                        if(a[i]==2)c2++;
                    }
                    s+=c1*(n-1);     
                    s+=(c2*(c2-1))/2;
                    cout<<s<<endl;
                }
        /* Enter your code here. Read input from STDIN. Print output to STDOUT */   
        return 0;
    }