ZOJ Problem Set - 2947 Abbreviation

#include <iostream>
#include <string>
using namespace std;

int main()
{
	int T , N , temp , flag;
	char word[6];
	string s;
	cin>>T;
	while (T--)
	{
		flag = 1;
		cin>>N;
		for (int i = 0; i < N; i++)
		{
			cin>>s;
			word[i] = s[0];
		}
		cin>>temp;
		if (N != temp)
		{
			flag = 0;
			for (int i = 0; i < temp; i++) { //注意要添加次循环输入，才能符合输入的要求 
				cin>>s;
			}
		}
		else
		{
			for (int i = 0; i < N; i++)
			{
				cin>>s;
				if (s[0] != word[i])
				{
					flag = 0;
				}
			}
		}
		if (flag) cout<<"SAME"<<endl;
		else cout<<"DIFFERENT"<<endl;
	}
	
	return 0;
}