Cod sursa(job #2064229)

Utilizator ionanghelinaIonut Anghelina ionanghelina Data 11 noiembrie 2017 23:41:35
Problema Dreptunghiuri Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.14 kb
#include<bits/stdc++.h>
#define maxN 405
using namespace std;
int n,m,a,b,c,delta;
int dp[maxN][maxN];
long long sol;
int ps[10000005];
int main()
{
    freopen("dreptunghiuri.in","r",stdin);
    freopen("dreptunghiuri.out","w",stdout);

    scanf("%d%d",&n,&m);
    for(int i=1;i<=500;i++) ps[i*i]=i;
    for(int L=1;L<n;L++)
        for(int l=1;l<m;l++)
    {
        int ans=0;
        for(int k=0;k<L;k++)
        {

            a=1;
            b=-l;
            c=-k*k+L*k;
            delta=b*b-4*a*c;
            if(delta<0) continue;

            if(delta==0)
            {
                int x=-b/(2*a);
                if(!(b%(2*a)) && x>=0 && x<l) ans++;
            }
                else
            if(delta>0)
            {
                int rp=ps[delta];
                if(!rp) continue;
                int x1=(-b-rp)/(2.0*a);
                int x2=(-b+rp)/(2.0*a);
                if(!((-b-rp)%(2*a)) && x1>=0 && x1<l) ans++;
                if(!((-b+rp)%(2*a)) && x2>=0 && x2<l) ans++;
            }
        }
        sol=sol+1LL*(n-L)*(m-l)*ans;
    }
    printf("%lld\n",sol);
    return 0;
}