update 002.c.  add array malloc part