牛客 Wannafly 挑战赛专题

Wannfly 27

A-灰魔法师：题目链接

emmm......此题......只能说我太弱了，一个签到题硬是看着题解做的...

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn = 1e6+1000;
int n;
ll a[maxn];
ll ans=0;
int main()
{
	cin >> n;
	int x;
	memset(a,0,sizeof(a));
	for(ll i=1;i<=n;i++){
        cin>>x;
        for(int j=1;j<500;j++){
            if(j*j>x)
                ans+=a[j*j-x];
        }
        a[x]++;
	}
	cout<<ans<<endl;
    return 0;
}

B-紫魔法师：题目链接

这是一个找规律+dfs的题，由于每条边最多被包含于一个环，所以不可能出现这种图：

这样的话，规律很容易找到：当边数为偶数时，最少颜色数是2，边数为奇数，最少颜色数为3。接下来就使用dfs搜索判断这是不是一个二分图，（因为二分图的判定过程就是：能否用黑白两色染色成功）。若是二分图，输出2；否则输出3。

#include<iostream>
#include<vector>
#include<string.h>
using namespace std;
typedef long long ll;
const int maxn=200000+10;
int n,m;
vector<int>vec[maxn];
int color[maxn];
bool vis[maxn];
int flag=0;
void dfs(int u,int c)
{
    color[u]=c;
    vis[u]=true;
    for(int i=0;i<vec[u].size();i++){
        int v=vec[u][i];
        if(!vis[v]){
            vis[v]=true;
            dfs(v,-c);
        }
        else if(color[v]==color[u]){
            flag=1;
            return;
        }
    }
}
int main()
{
    cin>>n>>m;
    int u,v;
    for(int i=1;i<=m;i++){
        cin>>u>>v;
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    memset(vis,0,sizeof(vis));
    dfs(1,1);
    if(flag)
        cout<<3<<endl;
    else
        cout<<2<<endl;
    return 0;
}