bzoj3545/bzoj3551 [ONTAK2010]Peaks/Peaks加强版

传送门：bzoj bzoj

wdnmd为什么加强版不是权限题原题却是啊

3545: [ONTAK2010]Peaks

Time Limit: 10 Sec Memory Limit: 128 MB
[Submit][Status][Discuss]

Description

在Bytemountains有N座山峰，每座山峰有他的高度h_i。有些山峰之间有双向道路相连，共M条路径，每条路径有一个困难值，这个值越大表示越难走，现在有Q组询问，每组询问询问从点v开始只经过困难值小于等于x的路径所能到达的山峰中第k高的山峰，如果无解输出-1。

Input

第一行三个数N，M，Q。
第二行N个数，第i个数为h_i
接下来M行，每行3个数a b c，表示从a到b有一条困难值为c的双向路径。
接下来Q行，每行三个数v x k，表示一组询问。

Output

对于每组询问，输出一个整数表示答案。

Sample Input

10 11 4

1 2 3 4 5 6 7 8 9 10

1 4 4

2 5 3

9 8 2

7 8 10

7 1 4

6 7 1

6 4 8

2 1 5

10 8 10

3 4 7

3 4 6

1 5 2

1 5 6

1 5 8

8 9 2

Sample Output

6

1

-1

8

HINT

【数据范围】

N<=10^5, M,Q<=5*10^5，h_i,c,x<=10^9。

果然还是太年轻了居然被这种水平的题吊打。。。

简短思路：

首先来一次kruskal

很明显不在树上的边都用不上

然后这时候就出现了kruskal重构树

随便挂个讲得好的博客

拼出来重构树之后用线段树合并或是主席树都行

以下通用代码(只差一排注释)

  1 #include<cstdio>
  2 #include<algorithm>
  3 #include<cstring>
  4 using namespace std;
  5 const int N=200011,inf=0x7f7f7f7f,M=500011;
  6 template<typename TP>inline void read(TP &kk)
  7 {
  8     TP ret=0,f=1;char ch=getchar();
  9     while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
 10     while(ch>='0'&&ch<='9'){ret=ret*10+ch-'0';ch=getchar();}
 11     kk=ret*f;
 12 }
 13 int n,m,Q;
 14 struct shino
 15 {
 16     int x,y,w;
 17     bool operator < (const shino &a)const
 18     {return w<a.w;}
 19 }sl[M];
 20 int snr;
 21 #define lson(x) (son[x][0])
 22 #define rson(x) (son[x][1])
 23 struct sgt
 24 {
 25     int c[N*36],son[N*36][2],cnt;
 26     void ins(const int x,int &px,int pl,int pr)
 27     {
 28         if(!px) px=++cnt;
 29         c[px]++;
 30         if(pl==pr) return;
 31         int mi=pl+pr>>1;
 32         if(x<=mi) ins(x,lson(px),pl,mi);
 33         else ins(x,rson(px),mi+1,pr);
 34     }
 35     int merge(const int px,const int py)
 36     {
 37         if(!px||!py) return px+py;
 38         int pz=++cnt;
 39         c[pz]=c[px]+c[py];
 40         lson(pz)=merge(lson(px),lson(py));
 41         rson(pz)=merge(rson(px),rson(py));
 42         return pz;
 43     }
 44     int query(int k,const int px,int pl,int pr)
 45     {
 46         if(pl==pr) return pl;
 47         int mi=pl+pr>>1;
 48         if(k<=c[lson(px)]) return query(k,lson(px),pl,mi);
 49         else return query(k-c[lson(px)],rson(px),mi+1,pr);
 50     }
 51 }tr;
 52 int rt[N];
 53 int h[N];
 54 int f[N],fa[N][20],val[N],son[N][2];
 55 int find(int x)
 56 {
 57     if(f[x]!=f[f[x]]) return f[x]=find(f[x]);
 58     return f[x];
 59 }
 60 void kksk()//k,k(ru)sk(al)... 无端玩梗的rkk是屑
 61 {
 62     snr=n,val[0]=inf;
 63     for(int i=1;i<=m;i++)
 64     {
 65         int x=find(sl[i].x),y=find(sl[i].y);
 66         if(x==y) continue;
 67         f[x]=f[y]=fa[x][0]=fa[y][0]=++snr;
 68         val[snr]=sl[i].w;
 69         son[snr][0]=x,son[snr][1]=y;
 70     }
 71 }
 72
 73 int saki[N];
 74 int main()
 75 {
 76     read(n),read(m),read(Q);
 77     for(int i=1;i<=n;i++)
 78         read(sl[i].w),sl[i].x=i;
 79     sort(sl+1,sl+1+n);
 80     for(int i=1;i<=n;i++)
 81     {
 82         if(sl[i].w!=sl[i-1].w) saki[++snr]=sl[i].w;
 83         h[sl[i].x]=snr;
 84     }
 85     for(int i=1;i<=m;i++)
 86         read(sl[i].x),read(sl[i].y),read(sl[i].w);
 87     sort(sl+1,sl+1+m);
 88     for(int i=1;i<=n*2;i++) f[i]=i;
 89     kksk();
 90     for(int i=1;i<=19;i++)
 91         for(int x=1;x<=snr;x++)
 92             fa[x][i]=fa[fa[x][i-1]][i-1];
 93     for(int i=1;i<=n;i++) tr.ins(h[i],rt[i],1,n);
 94     for(int i=n+1;i<=n*2;i++) rt[i]=tr.merge(rt[lson(i)],rt[rson(i)]);
 95     int lans=0,xi,vi,ki;
 96     while(Q--)
 97     {
 98         read(xi),read(vi),read(ki);
 99         xi^=lans,vi^=lans,ki^=lans;//前后就差个这行
100         for(int i=19;i>=0;i--)
101         {
102             if(val[fa[xi][i]]<=vi) xi=fa[xi][i];
103         }
104         if(tr.c[rt[xi]]<ki){puts("-1");lans=0;}
105         else printf("%d\n",lans=saki[tr.query(tr.c[rt[xi]]-ki+1,rt[xi],1,n)]);
106     }
107     return 0;
108 }