牛客周赛8题解

A:小美的排列询问

小模拟：用一个map来记录数字的位置，在后面判断两个数的abs（mp[a]-mp[b]）是不是等于1就可以

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
char c[1001][1001];
int val[N];//每个点的权值 
vector<int>g[N];
bool vis[N];
int check(int x,int y)
{
	ll p=x*y;
	int u=(int)sqrt(p);
	if(u*u==p)return 1;
	return 0;
 } 
int dfs(int u,int fa)
{
	int cnt=0;
	for(auto v:g[u])
	{
		if(v==fa)continue;
		int w=dfs(v,u);
		if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
		{
			//搜vu两个点 
			vis[u]=vis[v]=1;
			cnt+=2;
		}
		cnt+=w; 
		
	}
	return cnt;
}
void cf()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>val[i];
	}
	for(int i=1;i<=n-1;i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	for(int i=1;i<=n;i++)
	{
		if(g[i].size()==1){
			int ans=dfs(i,-1);
			cout<<ans<<endl;
			break;
		}
	}
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t;
	t=1;
	while(t--){
		cf();
	}
	 
	
	 
	 
}

B题：小美走公路

也是一道模拟题不过终点可能在起点前面这是一个坑点

所有x-y的距离不可以直接用for循环去遍历

while(x!=y){
        ans+=a[x];
       x=(x+1)%n;
    }

这里开了一个while，每次取模的话可以避免x套圈走

然后每种路径都有两种答案一种是x-y的，一种是反向走的

如图所示：所以答案一种是ans一种是sum-ans

我们两个取min就可以了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
ll a[N];
void cf()
{
    ll n;
    cin>>n;
   ll sum=0;
    for(int i=0;i<n;i++){
        cin>>a[i];
        sum+=a[i];
    }
   ll x,y;
    cin>>x>>y;
    x--;
    y--;
    ll ans=0;
    while(x!=y){
        ans+=a[x];
       x=(x+1)%n;
    }
    cout<<min(ans,sum-ans);
    
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t;
    t=1;
    while(t--)
    {
        cf();
    }
}

 C题：小美的排序构造：

我们可以很清楚的知道比如5：

我们可以构造出来：1 5 2 4 3

所以我们可以把每两个数之和构造n+1和n+2这样极差就是1

因此每次取出来一个数我们可以输出i和(n+1)-i

i的范围是在1-（n+1）/2这里的n+1是上取整操作

还有一个点是要注意奇偶

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int main(){
    int n;
    cin>>n;
    
    ll q=0;
    q=(n+1)/2;
    ll ans=n+1;
    if(n&1)
    {
    for(int i=1;i<=q;i++){
     if(i<=q-1){
         cout<<i<<" "<<ans-i<<" ";
     }   
        else
        {
            cout<<i;
        }
    }
    }
    else
    {
        for(int i=1;i<=q;i++)
        {
            cout<<i<<" "<<ans-i<<" ";
        }
    }
    
}

D题：小美的树上染色

有一种贪心的做法是直接找一个叶子然后从叶子开始去模拟遍历

以叶子为父节点一层一层去遍历

这里的check函数是判断乘积是不是完全平方数

vis数组是判断这个点有没有染色

dfs函数的意思是以u为节点fa为u的父节点向下搜的答案

所以w就是以u为父节点的答案

然后我们只需要再判断u和v能不能染色然后把两种答案相加就是总答案

int dfs(int u,int fa)
{
	int cnt=0;
	for(auto v:g[u])
	{
		if(v==fa)continue;
		int w=dfs(v,u);
		if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
		{
			//搜vu两个点 
			vis[u]=vis[v]=1;
			cnt+=2;
		}
		cnt+=w; 
		
	}
	return cnt;
}

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
char c[1001][1001];
int val[N];//每个点的权值 
vector<int>g[N];
bool vis[N];
int check(int x,int y)
{
	ll p=x*y;
	int u=(int)sqrt(p);
	if(u*u==p)return 1;
	return 0;
 } 
int dfs(int u,int fa)
{
	int cnt=0;
	for(auto v:g[u])
	{
		if(v==fa)continue;
		int w=dfs(v,u);
		if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
		{
			//搜vu两个点 
			vis[u]=vis[v]=1;
			cnt+=2;
		}
		cnt+=w; 
		
	}
	return cnt;
}
void cf()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>val[i];
	}
	for(int i=1;i<=n-1;i++)
	{
		int u,v;
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	for(int i=1;i<=n;i++)
	{
		if(g[i].size()==1){
			int ans=dfs(i,-1);
			cout<<ans<<endl;
			break;
		}
	}
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t;
	t=1;
	while(t--){
		cf();
	}
	 
	
	 
	 
}