经过测试,数据中存在 k=1k=1k=1 的情况。
#include<iostream> #include<cassert> int T,n,k; main(){ std::cin>>T;while(T--){ std::cin>>n>>k; assert(k>=2); for(int i=1,u,v;i<n;i++)std::cin>>u>>v; for(int i=1,x;i<=n;i++)std::cin>>x; } }