【Codeforces】Common Subsequence （map）

题目链接

思路：

这道题总的来说并不难，就是查找是否有相同元素。

• 若有输出YES，然后输出随意一个相同的元素（默认第一个相同元素，简单）；
• 若无输出NO；

但是起初我是利用两个数组来写的，在没找到第一个相同元素时，b数组每输入一个元素则遍历一次a数组查找是否有相同元素，时间复杂度太大。但是后来写完发现利用map直接把元素当作下标存入会更加迅速，节省了时间。

AC（两个数组）：

#include<iostream>
using namespace std;
int main()
{
	int t;
	cin >> t;
	while(t--)
	{
		int n,m;
		cin >> n >> m;
		int a[n],b[m],c,k=0;
		for(int i=0;i<n;i++)
			cin >> a[i];
		for(int i=0;i<m;i++)
		{
			cin >> b[i];
			if(k==0)
			{
				for(int j=0;j<n;j++)
				{
					if(b[i]==a[j])
					{
						c=b[i];
						k=1;
						break;
					}
				}
			}
		}
		if(k==0)
			cout<< "NO" << endl;
		else
			cout<<"YES" << endl << 1 << " " << c << endl;	
	}
}

代码（map）：

#include<iostream>
#include<map> 
using namespace std;
int main()
{
	
	int t;
	cin >> t;
	while(t--)
	{
		int n,m;
		cin >> n >> m;
		map<int ,int > p;
		int k,flag=-1;
		for(int i=0;i<n;i++)
		{
			cin >> k;
			p[k]++;
		}
		for(int i=0;i<m;i++)
		{
			cin >> k;
			if(p[k]&&flag==-1)
				flag=k;
		}
		if(flag==-1)
			cout<< "NO" << endl;
		else
			cout<<"YES" << endl << 1 << " " << flag << endl;	
	}
}