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

链接牛客网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;
    }
};


在这里插入图片描述

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页