Cod sursa(job #2293216)

Utilizator RaduXD1Nicolae Radu RaduXD1 Data 30 noiembrie 2018 17:38:59
Problema Dreptunghiuri Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include<fstream>
#include<cstring>
using namespace std;
ifstream fin("dreptunghiuri.in");
ofstream fout("dreptunghiuri.out");
int i,f[160001],v[160001],n,m,h,w,a,delta,sol,ss,aux,c1,c2;

int main()
{
    for(i=0;i<=400;i++)
        f[i*i]=i,v[i*i]=1;
    fin>>m>>n;
    if(m<n) swap(m,n);
    for(h=2;h<=n;h++)
        for(w=2;w<=m;w++)
        {
            sol=1;w--;h--;
            for(a=1;a<h;a++)
            {
                delta=w*w-4*a*(h-a);
                if(!v[delta]) continue;
                aux=f[delta];c1=(w-aux)/2;c2=(w+aux)/2;
                if(c1==c2&&w%2==0) sol--;
                if(!((w-aux)%2)&&c1>0&&c1<w) sol++;
                if(!((w+aux)%2)&&c2>0&&c2<w) sol++;
            }
            h++;w++;
            ss+=sol*(n-h+1)*(m-w+1);
        }
    fout<<ss;
    return 0;
}