পাইথনে ভেরটেক্সের ডিগ্রি থেকে গ্রাফ নাকি ট্রি নির্ণয় করা

ভেরটেক্সগুলোর ডিগ্রির লিস্ট দেওয়া থাকলে, সেটি কি একটি ট্রি (Tree) নাকি সাধারণ গ্রাফ (Graph) তা নির্ণয় করতে হবে। ট্রির জন্য ডিগ্রি যোগফল = ২ × (ভেরটেক্স সংখ্যা - ১) হবে।

উদাহরণ

ইনপুট: deg = [2, 2, 3, 1, 1, 1]

আউটপুট: Tree

সূত্র

ট্রি হলে sum(deg) == 2 * (n - 1) হবে, যেখানে n = ভেরটেক্স সংখ্যা।

উদাহরণ কোড

def solve(deg):
    vert = len(deg)
    deg_sum = sum(deg)
    if 2 * (vert - 1) == deg_sum:
        return 'Tree'
    return 'Graph'

deg = [2, 2, 3, 1, 1, 1]
print(solve(deg))  # Output: Tree

ইনপুট

[2, 2, 3, 1, 1, 1]

আউটপুট

Tree