Codeforces Round #474 F. Pathwalks【主席树】

题目链接

题意

给出一个 n n n 个结点的无向图图，找出边权递增且边的编号（即输入顺序）递增的路径的最长长度

题解

对于结点 X X X，用线段树维护所有能到达 X X X 的边中边权在 [ L , R ] [L,R] [L,R] 区间内最长的路径，一共需要 n n n 颗线段树，所以采用主席树

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
#define mid ((l+r)>>1)

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 1e5+10;
struct node{
	int l,r;
	int w;
}hjt[N*30];
int cnt,root[N];
void update(int &now,int l,int r,int pos,int w){
	if(!now)now=++cnt;
	hjt[now].w=max(hjt[now].w,w);
	if(l==r)return;
	if(pos<=mid)update(hjt[now].l,l,mid,pos,w);
	else update(hjt[now].r,mid+1,r,pos,w);
}
int query(int now,int l,int r,int x,int y){
	if(x<=l&&r<=y)return hjt[now].w;
	int ans=0;
	if(x<=mid&&hjt[now].l)ans=max(ans,query(hjt[now].l,l,mid,x,y));
	if(y>mid&&hjt[now].r)ans=max(ans,query(hjt[now].r,mid+1,r,x,y));
	return ans;
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int n,m,ans=0,now; cin>>n>>m;
	for(int i=1;i<=m;i++){
		int u,v,w; cin>>u>>v>>w;
		now=query(root[u],0,100000,0,w-1)+1;
		update(root[v],0,100000,w,now);
		ans=max(now,ans);
	}
	cout<<ans<<endl;
}

标签：hjt,include,Pathwalks,Codeforces,int,474,ans,now,define
来源： https://blog.csdn.net/qq_50554695/article/details/120824482