编程题:首尾相连的字符串

给定一个字符串数组（字符串长度和数组的长度均大于1且小雨1024），所有字符均为大写字母。请问，给定的字符串数组是否能通过更换数组中元素的顺序，从而首尾相连，形成一个环，环上相邻字符串首位衔接的字符相同

输入描述：
一行输入，空格分隔，表示字符串数组
输出描述：
一行输出，返回true或者false，表示是否可以形成环
输入1

CAT TIGER REC

输出1

true

输入2

CAT RPC

输出2

false

---

思路，设置一个bool，把第一个位置固定住，然后在fun里面寻找和k-1能首尾相连的字符串，找到了之后和第k个位置交换。。。。。。交换之后再寻找第k+1个位置。。。。。。。。然后回溯，把第k个位置交换回来，看看有没有其他能和第k个位置首位相连的。

 

#include <cstring>
#include <string>
#include <iostream>
#include <vector>
using namespace std;
bool flag = false;
void swap(vector<string> &arr, int s, int i)
{
	string tmp = arr[s];
	arr[s] = arr[i];
	arr[i] = tmp;
	return;
}
void fun(vector<string> arr, int n, int k)
{
	if(k==n)
	{
		if(arr[k-1].back()==arr[0][0])
			flag = true;
		return;
	}
	for (int i = k; i < n; ++i) {
		if((arr[k-1].back()==arr[i][0]))
		{
			swap(arr,k,i);
			fun(arr,n,k+1);
			swap(arr,k,i);
		}
	}
}
bool judge(vector<string> arr, int n)
{
	fun(arr, n, 1);
	if(flag) return true;
	else return false;
}
int main()
{
	vector<string> A;
	while(true)
	{
		string s;
		cin>>s;
		A.push_back(s);
		char c = cin.get();
		if(c=='\n')
			break;
	}
	if(A.size()<1)
	{
		cout<<"false"<<endl;
		return 0;
	}
	if(A.size()==1)
	{
		if(A[0].back()==A[0][0])
		{
			cout<<"true"<<endl;
		}
		else
			cout<<"false"<<endl;
		return 0;
	}
	if(judge(A,A.size())) cout<<"true"<<endl;
	else cout<<"false"<<endl;
	return 0;
}