Cod sursa(job #338652)

Utilizator GavrilaVladGavrila Vlad GavrilaVlad Data 6 august 2009 13:11:59
Problema Dreptunghiuri Scor 100
Compilator cpp Status done
Runda Lista lui wefgef Marime 1.75 kb
#include <stdio.h>
#include <math.h>

using namespace std;

#define maxn 1001

long n, m, i, j, k, w, h, a, b, c, x, d, sol, nr, rad;
long long tot;
long v[maxn*maxn];

int main()
{
    freopen("dreptunghiuri.in", "r", stdin);
    freopen("dreptunghiuri.out", "w", stdout);
    scanf("%d%d", &n, &m);
    n--;
    m--;
    for(i=1; i<maxn; i++)
    {
        v[i*i]=i;
    }
    for(i=1; i<=n; i++)
    {
        for(j=1; j<=m; j++)
        {
            nr=1;
            for(k=1; k<j; k++)
            {
                a=1;
                b=-i;
                c=k*(j-k);
                d=b*b-4*a*c;
                if(d<0) continue;
                if(d==0 && (-b%(2*a))==0)
                {
                    sol=-b/(2*a);
                    if(sol>=0 && sol<=i)
                    {
                        nr++;
                    }
                }
                else
                {
                    rad=v[d];
                    if(rad!=0)
                    {
                        if((-b+rad)%(2*a)==0)
                        {
                            sol=(-b+rad)/(2*a);
                            if(sol>=0 && sol<=i)
                            {
                                nr++;
                            }
                        }
                        if((-b-rad)%(2*a)==0)
                        {
                            sol=(-b-rad)/(2*a);
                            if(sol>=0 && sol<=i)
                            {
                                nr++;
                            }
                        }
                    }
                }
            }
            tot+=nr*(n-i+1)*(m-j+1);
        }
    }
    printf("%lld\n", tot);
    return 0;
}