优化这段代码:#include<bits/stdc++.h> using namespace std; #define pb push_back const int MAXN=2e5+5; int T,n; int su,en[MAXN<<1],lt[MAXN<<1],id[MAXN<<1],hd[MAXN]; bool isok,vis[MAXN]; int s[MAXN]; void add(int u,int v,int w) { en[++su]=v,id[su]=w,lt[su]=hd[u],hd[u]=su; } void dfs(int x,int fa) { if(!isok) return; int f=0; for(int i=hd[x];i;i=lt[i]) { if(en[i]==fa) f=id[i]; else { dfs(en[i],x); s[x]+=s[en[i]]; } } s[x]++; if(s[x]==3) { vis[f]=1; s[x]=0; } else if(s[x]>3) isok=0; } void solve() { isok=1; su=0; for(int i=1;i<=n;++i) hd[i]=s[i]=0; memset(vis,0,sizeof(vis)); scanf("%d",&n); for(int i=1;i<n;++i) { int u,v; scanf("%d%d",&u,&v); add(u,v,i),add(v,u,i); } if(n%3!=0) { puts("-1"); return; } dfs(1,0); if(!isok) { puts("-1"); } else { int cnt=0; for(int i=1;i<=n;++i) if(vis[i]) ++cnt; printf("%d\n",cnt); for(int i=1;i<=n;++i) if(vis[i]) printf("%d ",i); printf("\n"); } } int main() { scanf("%d",&T); while(T--) solve(); return 0; }
时间: 2024-04-08 20:33:30 浏览: 127
优化代码的方法有很多,以下是一些常见的优化建议:
1. 头文件引用:只包含实际需要的头文件,避免使用 `bits/stdc++.h`,它包含了大量的标准库头文件,会增加编译时间和代码冗余。
2. 常量定义:避免使用 `#define` 宏定义,可以使用 `const` 关键字定义常量。
3. 循环边界:在循环中使用 `<` 来替代 `<=`,可以减少不必要的比较操作。
4. 数组初始化:可以使用 `std::fill` 或 `memset` 函数来初始化数组,避免逐个元素赋值。
5. 输入输出优化:可以使用快速的输入输出函数(如 `scanf` 和 `printf`)来提高输入输出效率。
6. 数据结构优化:考虑使用更高效的数据结构,例如使用 `std::vector` 替代数组,使用 `std::unordered_map` 替代 `std::map`。
下面是优化后的代码示例:
```cpp
#include <iostream>
#include <vector>
#include <cstring>
#include <unordered_map>
const int MAXN = 2e5 + 5;
struct Edge {
int v, w;
};
int T, n;
int su, en[MAXN << 1], lt[MAXN << 1], id[MAXN << 1], hd[MAXN];
bool isok, vis[MAXN];
int s[MAXN];
void add(int u, int v, int w) {
en[++su] = v;
id[su] = w;
lt[su] = hd[u];
hd[u] = su;
}
void dfs(int x, int fa) {
if (!isok) return;
int f = 0;
for (int i = hd[x]; i; i = lt[i]) {
if (en[i] == fa) f = id[i];
else {
dfs(en[i], x);
s[x] += s[en[i]];
}
}
s[x]++;
if (s[x] == 3) {
vis[f] = true;
s[x] = 0;
} else if (s[x] > 3)
isok = false;
}
void solve() {
isok = true;
su = 0;
for (int i = 1; i <= n; ++i)
hd[i] = s[i] = 0;
memset(vis, false, sizeof(vis));
scanf("%d", &n);
for (int i = 1; i < n; ++i) {
int u, v;
scanf("%d%d", &u, &v);
add(u, v, i), add(v, u, i);
}
if (n % 3 != 0) {
puts("-1");
return;
}
dfs(1, 0);
if (!isok) {
puts("-1");
} else {
int cnt = 0;
for (int i = 1; i <= n; ++i)
if (vis[i]) ++cnt;
printf("%d\n", cnt);
for (int i = 1; i <= n; ++i)
if (vis[i]) printf("%d ", i);
printf("\n");
}
}
int main() {
scanf("%d", &T);
while (T--)
solve();
return 0;
}
```
请注意,这只是对代码的简单优化,具体的优化策略还要根据实际情况进行分析和调整。
阅读全文