a291: nAnB problem

內容 ：
    我們常用數字密碼鎖來保護重要的東西，但要是不小心忘了密碼麻煩就大了！
    以四位數字的密碼鎖為例，我們最多要嘗試10^4=10000次才能解鎖。這時候要是
    有辦法知道目前嘗試的密碼錯了幾個字，那解鎖的速度就快多了。請寫一個程式，
    可以判斷每組數字跟正確答案差了幾個字。

輸入說明 ：
    多筆輸入。
    第一行有四個介於0-9之間的數字，代表正確的密碼
    第二行有一個整數n，1<=n<=10000，代表接下來嘗試n組密碼
    接下來有n行，每行有四個介於0-9之間的數字，每行各代表一組嘗試的密碼。

輸出說明 ：
    輸出n行。
    對於每組嘗試的密碼，若有p個數字的值正確，且在正確的位子上，
    另外有q個數字的值正確，但不在正確的位子上，
    輸出pAqB。
    範例見測資。

範例輸入 ：
1 2 3 4
4
1 1 4 5
1 2 4 3
1 1 4 4
4 3 2 1

1 1 1 5
4
1 1 1 1
0 9 2 8
1 5 2 3
1 1 5 1

範例輸出 ：
1A1B
2A2B
2A0B
0A4B
3A0B
0A0B
1A1B
2A2B

程式碼 ：

#include<stdio.h>
int main()
{
    int a,b,i,j,n,ans[4],input[4],check[4];
    while(scanf("%d %d %d %d",&ans[0],&ans[1],&ans[2],&ans[3])==4)
    {
        scanf("%d",&n);
        while(n--)
        {
            a = b = 0;
            for(i=0; i<4; i++)
            {
                scanf("%d",&input[i]);
                if(ans[i] == input[i])
                {
                    a++;
                    check[i] = 1;
                }
                else
                    check[i] = 0;
            }
            for(i=0; i<4; i++)
                if(check[i] != 1)
                    for(j=0; j<4; j++)
                        if(check[j]==0 && (input[i] == ans[j]) && j!=i)
                        {
                            b++;
                            check[j] = 2;
                            break;
                        }
            printf("%dA%dB\n",a,b);
        }
    }
    return 0;
}

http://zerojudge.tw/ShowProblem?problemid=a291