# Shaurya Baranwal Homework #8 due February 18, 2024

#-------------------------------------
# Functions needed to complete parts | 
#-------------------------------------

Fqn:=proc(q,n) local S,a,v:

  option remember:

  if n=0 then
    RETURN({[]}):
  fi:

  S:=Fqn(q,n-1):

  {seq(seq([op(v),a],a=0..q-1), v in S)}:

end:

#------------------------
# Part 1 - WtEnumerator |
#------------------------

wt := proc(v) local weight, i:

  weight := 0:

  for i from 1 to nops(v) do:
    if v[i] <> 0 then:
      weight := weight + 1:
    fi:
  od:

end:

WtEnumerator := proc(q,M,t) local C, v:

  C := Fqn(q, nops(M[1])):
  add(seq(t^wt(v), v in 'C')):

end: