Cod sursa(job #1767468)

Utilizator vlad2004SuntUnNanoTub vlad2004 Data 29 septembrie 2016 11:02:03
Problema Factorial Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include <fstream>
using namespace std;

ifstream in("nrtri.in");
ofstream out("nrtri.out");
int v[500001];
int v2[500001];
void merge(int l,int r)
{
    if(l==r)
        return ;
    else
    {
        merge(l,(l+r)/2);
        merge((l+r)/2+1,r);
        int i=l,j=(l+r)/2+1;
        int k=l;
        while(i<=(l+r)/2 || j<=r)
        {
            if(j>r || (i<=(l+r)/2 && v[i]<=v[j]))
            {
                v2[k]=v[i];
                k++;
                i++;
            }
            else
            {
                v2[k]=v[j];
                k++;
                j++;
            }
        }
        for(i=l; i<=r; i++)
        {
            v[i]=v2[i];
        }

    }
}
int main()
{
    int n,cnt=0,j,i,x,k;
    in>>n;
    for(i=1; i<=n; i++)
    {
        in>>v[i];
    }
    merge(1,n);
    int n2=1;
    while(n2*2<=n)
    {
        n2*=2;
    }
    for(i=1; i<=n; i++)
    {
        for(j=i+1; j<=n; j++)
        {
            int sol;
            sol=j;
            for(int k=n2; k>=1; k/=2)
            {
                if(sol+k<=n && v[sol+k]<=v[i]+v[j])
                {
                    sol=sol+k;
                }
            }
            cnt=cnt+sol-j;
        }
    }
    out<<cnt;
    return 0;
}