3804 Query on a tree

五一给这道题折磨得好痛苦，这么恶心的题目再也不想写第二遍

 

题目的思路很简单，求节点1到某个节点x路径上不大于y的最大边权

 

离线操作，mutiset里存的是边权的负值，用lower_bound可以方便求得集合里不大于y的最大值

题目恶心的地方是不能用递归的深搜， 只能写非递归的版本，第一次写非递归的深搜，写了挺久，但是一A挺开心的

 

#include <cstdio> #include <set> #include <vector> #include <map> using namespace std; #define maxn 100001 struct Node{ int v,w; // v 下个节点的编号, w边权 Node(){} Node(int av, int aw):v(av),w(aw){} }; struct QueNode{// 因为是离线操作，所以需要这个结构体来记录输出的顺序 int v; int bound; }; struct DPSType { // 非递归深搜要用到的结构体 int v, cid; int insw; }; vector<Node> adj[maxn]; map<int, int> qmp[maxn]; vector<QueNode> qseq; multiset<int> ms; int vis[maxn]; int t,n,m; DPSType vsta[maxn]; void solve() { int dpt = 0; vsta[dpt].cid = -1; vsta[dpt].v = 1; vis[vsta[dpt].v] = 1; while(dpt != -1) { const DPSType cnode = vsta[dpt]; if(cnode.cid + 1 < adj[cnode.v].size()) { // 如果这一层还没超出边界，就继续查询下去 vsta[dpt].cid = cnode.cid + 1; int cu = cnode.v; int nv = adj[cu][vsta[dpt].cid].v; if(!vis[nv]) { int insval = -(adj[cu][vsta[dpt].cid].w); ms.insert( insval ); ++dpt; vsta[dpt].v = nv; vsta[dpt].insw = insval; vis[nv] = 1; for(map<int, int>::iterator iter = qmp[nv].begin(); iter != qmp[nv].end(); ++iter) { multiset<int>::iterator rs = ms.lower_bound( -(iter->first) ); if(rs!=ms.end()){ iter->second = -(*rs); } } } } else { // 超出了边界，回溯 vsta[dpt].cid = -1; multiset<int>::iterator epos = ms.find(vsta[dpt].insw); if(epos!=ms.end()) ms.erase(epos); --dpt; } } } void ini(int n) { ms.clear(); qseq.clear(); for(int i = 0; i <= n; ++i) { adj[i].clear(); qmp[i].clear(); vsta[i].cid = -1; vis[i] = 0; } } int main () { scanf("%d", &t); while(t--) { scanf("%d", &n); ini(n); for(int i = 1; i <= n; ++i) adj[i].clear(); for(int i = 0; i < n-1; ++i) { int u,v,w; scanf("%d%d%d", &u, &v, &w); adj[u].push_back(Node(v,w)); adj[v].push_back(Node(u,w)); } scanf("%d", &m); QueNode on; for(int i = 0; i < m; ++i) { int v, b; scanf("%d%d", &v, &b); on.v = v; on.bound = b; qseq.push_back(on); qmp[v].insert(pair<int, int>(b, -1)); } solve(); for(size_t i = 0; i < qseq.size(); ++i) { QueNode tmp = qseq[i]; printf("%d/n", qmp[tmp.v].find(tmp.bound)->second ); } } return 0; }