L1-062 幸运彩票 (15 分)

彩票的号码有 6 位数字，若一张彩票的前 3 位上的数之和等于后 3 位上的数之和，则称这张彩票是幸运的。本题就请你判断给定的彩票是不是幸运的。

输入格式：

输入在第一行中给出一个正整数 N（≤ 100）。随后 N 行，每行给出一张彩票的 6 位数字。

输出格式：

对每张彩票，如果它是幸运的，就在一行中输出 You are lucky!；否则输出 Wish you good luck.。

输入样例：

2
233008
123456

输出样例：

You are lucky!
Wish you good luck.

答案

#include<bits/stdc++.h>
#include<cmath>
using namespace std;
int main()
{
	int n;
	cin>>n;
	int z[101]={0};
	for(int i=0;i<n;i++)
	{
		long long a;
		cin>>a;
		int total_a=0;
		int total_b=0;
		int j=3;
		while(j)
		{
			total_a=total_a+a%10;
			a=a/10;
			j--;
		}
		j=3;
		while(j)
		{
			total_b=total_b+a%10;
			a=a/10;
			j--;
		}
		if(total_a==total_b)
			z[i]=1;
	}
	for(int i=0;i<n;i++)
	{
		if(z[i]==1)
			cout<<"You are lucky!"<<endl;
		else
			cout<<"Wish you good luck."<<endl;
	}
	
}