牛客小百月赛43

E满意的集合

题目链接

问题描述：

主要思路：

刚刚看到这个问题的时候觉得是先选出来是三的倍数的组合，然后再2的x次方判断选不选，但是发现根本就不能这么做。
看了dalao的思路，这个题目应该是用dp做

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
int a[20];
ll dp[20][10];
int main()
{
    dp[0][0]=1;
    for(int i=1;i<=9;i++) cin>>a[i];
    for(int i=1;i<10;i++) //循环9个数
    {
        for(int j=0;j<3;j++) //循环dp[][j]  //!!!j与k遍历顺序反过来也是一样的，都正确
        {
            for(int k=0;k<3;k++) //循环k*a[i],也就是数字i选取多少
            {
                int beishu=(a[i]-k+3)/3;
                dp[i][(k*i+j)%3]+=beishu*dp[i-1][j];
                //直接+就可以，因为选取第一组，与第二组是一样的都是选取了1组
                dp[i][(k*i+j)%3]%=mod;
            }
        }
    }
    cout<<dp[9][0]<<endl;
    return 0;
}

F全体集合

题目链接

问题描述：

主要思路：

本题在做的过程中就想到了是和两点间的距离奇偶有关的，但是当时想的是要分别计算每个点与各个点之间的距离才可以判断，这样的话时间复杂度太高了，但是比赛结束后看了别人的代码，发现其实只要随便选取一个点，然后利用bfs计算该点与其他点之间的距离，然后再判断奇偶就可以了

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int N=2e5+10;
typedef pair<int,int> pa;
vector<int> a[N];
vector<int> b;
bool book[N][10];
int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    while(m--)
    {
        int x,y;
        cin>>x>>y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    while(k--)
    {
        int x;
        cin>>x;
        b.push_back(x);
    }
    queue<pa> q;
    book[b[0]][0]=1;
    q.push({b[0],0});
    while(q.size())
    {
        pa now=q.front();
        q.pop();
        int nowx=now.first;
        int tmp=now.second;
        for(int i=0;i<a[nowx].size();i++)
        {
            int nowxx=a[nowx][i];
            int tmpp=tmp^1;
            if(book[nowxx][tmpp]) continue;
            book[nowxx][tmpp]=1;
            q.push({nowxx,tmpp});
        }
    }
    bool ans=1;
    for(int x:b)
    {
        if(!book[x][0])
        {
            ans=0;
            break;
        }
    }
    if(ans) cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
    return 0;
}