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