神奇的木头

感觉题目不难但是要分清情况，要有一个明确的思路。

#include<stdio.h>
int main()
{
    int t,n,k,a[10],i,s;
    scanf("%d",&t);
    while(t--)
    {
        s=0;
        scanf("%d",&n);
        for(i=0;i<n;i++)
            scanf("%d",&a[i]);
        scanf("%d",&k);
        if(k==0||k==1)
        {
            s=k+a[0];
            for(i=1;i<n;i++)
            {
                if(a[i]==1||a[i]==0||s==1||s==0)
                       s=s+a[i];
                else
                    s=s*a[i];
            }
        }
        else
        {
            s=k;
        for(i=0;i<n;i++)
        {
                if(a[i]==0||a[i]==1)
                    s=s+a[i];
                else
                    s=s*a[i];
             
        }
        }
        printf("%d\n",s);
    }
    return 0;
}