pat 1076. Wifi密码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int solve(char ch)
{
  switch(ch)
  {
    case 'A': return 1;break;  
    case 'B': return 2;break;
    case 'C': return 3;break;
    case 'D': return 4;break;
//    default : return 0;
  }  
} 
int main()
{
  int n ;
  char problems[17];//因为答案长度固定，开始的时候对于长度一直没处理好
  memset(problems,'0',sizeof(problems));
  scanf("%d",&n);
  getchar();
  int *password = (int*)malloc(n*sizeof(int));
  memset(password,0,sizeof(password));
  for(int j = 1;j <= n; j++)
  {
    for(int i =1;i<=16;i++)
    {
      scanf("%c",&problems[i]);
    }
    for(int k = 1; k <= 16 ;k+=1)
    {
      if(problems[k] == 'T')
      {
        int ans = solve(problems[k-2]);
        *(password+j) = ans;
        break;
      }
    }
  }
  for(int j = 1;j <= n;j++)
  {
    printf("%d",*(password+j));
  }
  return 0;
}