牛客网NC45、NC43-7.29-树、树

牛客网算法解析

最新推荐文章于 2024-07-06 11:09:29 发布

原创最新推荐文章于 2024-07-06 11:09:29 发布 · 486 阅读

0 ·

CC 4.0 BY-SA版权

每日一题专栏收录该内容

13 篇文章

订阅专栏

本文深入解析了牛客网上的两道算法题目：NC45和NC43，详细介绍了树形结构的遍历和反转操作，以及如何通过判断反转的奇偶性来解决实际问题。文章提供了Java和C++两种语言的实现代码，并对比了它们在牛客网平台上的运行情况。

链接：牛客网NC45
题意、输入、输出：
在这里插入图片描述
分析：树。。图比较好
思路：写就完了，但令人难受的是，为什么qi的值要判断。。
代码：

import java.util.*;
 
 
public class Solution {
    /**
     * 扩散
     * @param n int整型
     * @param m int整型
     * @param u int整型一维数组
     * @param v int整型一维数组
     * @param q int整型一维数组
     * @return int整型一维数组
     */
    public int[] solve (int n, int m, int[] u, int[] v, int[] q) {
        // write code here
        int[] res=new int[n];
        int[] crush=new int[n];
        for(int i=0;i<m;i++){
            if(q[i]<=0||q[i]>n)continue;
            res[q[i]-1]++;
            crush[q[i]-1]++;
        }
        for(int i=0;i<n-1;i++){
            res[u[i]-1]+=crush[v[i]-1];
            res[v[i]-1]+=crush[u[i]-1];
        }
        return res;
    }
}

NC43
链接：
牛客网NC43
题意、输入、输出：
在这里插入图片描述
分析：树的遍历，偶数次反转等于没有反转
思路：通过反转的奇偶来判断是否反转，只记录开始反转的根节点，统计所有反转后从根开始反转，如果一个节点要反转，将反转的信号向下传递。
代码：
一样的写法，c++能过，java不能，为啥。。。
java:

import java.util.*;
 
 
public class Solution {
    /**
     * 魔力转圈圈
     * @param n int整型
     * @param m int整型
     * @param l int整型一维数组
     * @param r int整型一维数组
     * @param k int整型一维数组
     * @return int整型一维数组
     */
    public int cnt=0;
    public void dfs(int x,int[] ans,boolean[] isRo,int[] l,int[] r){
        if(isRo[x]){
            if(r[x]!=0){
                isRo[r[x]-1]=!isRo[r[x]-1];
                dfs(r[x]-1,ans,isRo,l,r);
            }
            ans[cnt++]=x+1;
            if(l[x]!=0){
                isRo[l[x]-1]=!isRo[l[x]-1];
                dfs(l[x]-1,ans,isRo,l,r);
            }
        }else{
            if(l[x]!=0)dfs(l[x]-1,ans,isRo,l,r);
            ans[cnt++]=x+1;
            if(r[x]!=0)dfs(r[x]-1,ans,isRo,l,r);
        }
    }
    public int[] solve (int n, int m, int[] l, int[] r, int[] k) {
        // write code here
        boolean[] isRo=new boolean[n];
        Arrays.fill(isRo,false);
        for(int i=0;i<m;i++){
            isRo[k[i]-1]=!isRo[k[i]-1];
        }
        int[] ans=new int[n];
        dfs(0,ans,isRo,l,r);
        return ans;
    }

c++：

class Solution {
public:
    /**
     * 魔力转圈圈
     * @param n int整型 
     * @param m int整型 
     * @param l int整型vector 
     * @param r int整型vector 
     * @param k int整型vector 
     * @return int整型vector
     */
    void dfs(int x,vector<int>& ans,bool isRo[],vector<int>& l,vector<int>& r){
        if(isRo[x]){
            if(r[x]!=0){
                isRo[r[x]-1]=!isRo[r[x]-1];
                dfs(r[x]-1,ans,isRo,l,r);
            }
            ans.push_back(x+1);
            if(l[x]!=0){
                isRo[l[x]-1]=!isRo[l[x]-1];
                dfs(l[x]-1,ans,isRo,l,r);
            }
        }
        else {
            if(l[x]!=0)dfs(l[x]-1,ans,isRo,l,r);
            ans.push_back(x+1);
            if(r[x]!=0)dfs(r[x]-1,ans,isRo,l,r);
        }
    }
    vector<int> solve(int n, int m, vector<int>& l, vector<int>& r, vector<int>& k) {
        // write code here
        bool isRo[n];
        memset(isRo,false,sizeof(isRo));
        for(int i=0;i<m;i++){
            isRo[k[i]-1]=!isRo[k[i]-1];
        }
        vector<int> ans;
        dfs(0,ans,isRo,l,r);
        return ans;
    }
};