PAT甲级1047 Student List for Course

AC代码

#include<bits/stdc++.h>
using namespace std;
void change(char out[],int id)
{
  out[3]=id%10+'0';
  id/=10;
  for(int i=2;i>=0;i--)
  {
    out[i]=id%26+'A';
    id/=26;
  }
}
int main()
{
  int n,k,c,num;
  scanf("%d%d",&n,&k);
  vector<int> v[k+1];
  char name[4];
  for(int i=0;i<n;i++)
  {
    scanf("%s %d",name,&c);
    int id=0;
    for(int j=0;j<3;j++)
    {
      id=id*26+name[j]-'A';
    }
    id=id*10+name[3]-'0';
    for(int j=0;j<c;j++)
    {
      scanf("%d",&num);
      v[num].push_back(id);
    }
  }
  for(int i=1;i<=k;i++)
  {
    int size=v[i].size();
    sort(v[i].begin(),v[i].end());
    printf("%d %d\n",i,size);
    for(int j=0;j<size;j++)
    {
	  char out[4];
      change(out,v[i][j]);
      for(int k=0;k<4;k++) printf("%c",out[k]);
      printf("\n");
     }
  }
  return 0;
}