7-1 懂的都懂——2021 Robocom 世界机器人开发者大赛-本科组（初赛）【DFS+哈希表】

---

前言

这题主要考查DFS和哈希表。

看完提示的同学可以先再想一想如何解出这道题

---

一、题目

7-1懂得都懂（20分）

众所周知，在互联网上有很多话是不好直接说出来的，不过一些模糊的图片仍然能让网友看懂你在说什么。然而对这种言论依然一定要出重拳，所以请你实现一个简单的匹配算法。

现在我们采集了原图的一些特征数据，由 N 个小于 255 的非负整数组成，假设对于给定的若干张由 M**i 个同样小于 255 的非负整数组成的新图的特征数据，每个数据都可以由原图中任意四个不同数据的平均值计算而来，则称新图为原图的相似图片。对于给出的数据，请你判断是不是相似图片。

注意，不同数据指的并非是数据的值不同，而是不能取同一个数据多次。对于两个相同值的数据，如果给出两次，则可以取两次。

输入格式:
输入第一行是两个整数 N,K (1 ≤ N ≤ 50, 1 ≤ K ≤ 200)，表示采集的原图的特征数据个数和新图的张数。

接下来一行为 N 个小于 255 的非负整数，表示原图的特征数据。

最后的 K 行，每行第一个数是 Mi(1 ≤ Mi ≤ 200)，表示新图的特征数据个数。然后是 Mi 个小于 255 的非负整数，表示新图的特征数据。

输出格式:
对于每一张新图，如果为相似图片，则在一行中输出 Yes，否则输出 No。

输入样例:

5 3
4 8 12 20 40
3 11 16 19
3 12 16 19
10 11 11 11 11 11 11 11 11 11 11

输出样例:

Yes
No
Yes

---

二、解题过程

1.解题思路

（1）首先用DFS去遍历原图，将原图每四个元素相加得到的结果存到哈希表里。

（2）接着再去遍历新图的每个元素，判断是否在哈希表里，如果新图的全部元素都在哈希表里，则输出Yes，否则，输出No。

---

2.代码

#include<bits/stdc++.h>
using namespace std;
const int N = 51;
int n,k,m,x;
int a[N];
unordered_map<int,int> mp;
// j：上一层加入sum的元素下标
// num：sum当前所加元素的个数
// sum：元素之和
void dfs(int j,int num,int sum)
{
    // 如果元素数量达到四个就返回
    if (num==4) 
    {
        mp[sum]=1;
        return;
    }
    // i从j+1开始，因为j在上一层已经加过了
    for (int i=j+1;i<=n;i++)
        // 传入当前所加元素的下标，所加元素个数，元素之和
        dfs(i,num+1,sum+a[i]);
}
int main(){
    cin>>n>>k;
    // 输入原图的元素
    for (int i=1;i<=n;i++)
        cin>>a[i];
    // 将原图每四个元素相加得到的结果存到哈希表里
    dfs(0,0,0);
    while (k--)
    {
        cin>>m;
        bool flag=true;
        for (int i=1;i<=m;i++)
        {
            cin>>x;
            // 因为哈希表存的是总和，不是平均数，所以这里将新图每个元素乘4
            if (mp[x*4]) continue;
            flag=false;
        }
        if (flag) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}

---

如果觉得对你有所帮助的话，请点个免费的赞吧！