python统计方格数里面有多少正方形

题目

计算下图有多少正方形？

代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

def count(horizontal, vertical):
    
    _h = 0
    _c = 0
    while _h < horizontal:
        
        _v = 0

        while _v < vertical:
            
            if vertical - _v > horizontal - _h:
                _c += horizontal - _h
            else:
                _c += vertical - _v
            
            _v += 1

        _h += 1

    return _c

x = input("请输入横向单元格个数")
y = input("请输入纵向单元格个数")

print(count(int(x),int(y)))

思路：

1. 定义横向与纵向的轴
2. 按照一个顶点，顺着两个方向依次计算每一个小格子开始的正方形