题目大意:给你一个图,有一些操作要你实现。具体操作见题目。
解题思路:首先,我认真地读着题目,当我看到
这两行时,感觉有点不知所措。然而,
!!!
这充分说明了出题者脑洞之大。orz
当我历经千辛万苦写完代码打算测样例时,发现样例第11行出现了个操作9!而且还影响答案!相当于这题连样例都没有。
吐槽完了,回归正题。这道题不难看出,需要用线段树合并来做。并且在建立权值线段树之前,先得把数据离散化(数据<=1000000000)。
对于操作3和4,删除所有需要改变的节点,然后加进权值x里即可。
对于操作6,可以用$\log_{}ab=\log_{}a+\log_{}b$的方式进行比较。
其他操作随便搞搞就行了。
数组不要开太小,当然也不能开太大。
C++ Code:
#include<cstdio>
#include<algorithm>
#include<map>
#include<cmath>
#define N 6000000
using namespace std;
int m,n=0,mx=0,node=0,u,v,x,cnt;
int fa[400001],root[400001],ld[N],rd[N],mp[400001],d[N];
double lg[N]={0};
struct operation{
int c,a,b;
}ol[400001];
map<int,int>p;
inline void update(int k){
d[k]=d[ld[k]]+d[rd[k]];
lg[k]=lg[ld[k]]+lg[rd[k]];
}
int dad(int x){return fa[x]==x?x:(fa[x]=dad(fa[x]));}
void add(int& k,int l,int r,int x,int num){
if(!k)k=++node;
if(l==r){
d[k]+=num;
lg[k]+=num*log(mp[x]);
return;
}
int mid=l+r>>1;
if(x<=mid)add(ld[k],l,mid,x,num);else
add(rd[k],mid+1,r,x,num);
update(k);
}
inline int merge(int u,int v,int l,int r){
if(!u||!v)return u|v;
if(l==r){
d[u]+=d[v];
lg[u]+=lg[v];
return u;
}
int mid=l+r>>1;
ld[u]=merge(ld[u],ld[v],l,mid);
rd[u]=merge(rd[u],rd[v],mid+1,r);
update(u);
return u;
}
int query(int k,int l,int r,int L,int R){
if(L<=l&&r<=R)return d[k];
int m=l+r>>1,ans=0;
if(L<=m)ans+=query(ld[k],l,m,L,R);
if(m<R)ans+=query(rd[k],m+1,r,L,R);
return ans;
}
void del(int k,int l,int r,int L,int R){
if(!d[k])return;
if(l==r){
d[k]=0;
lg[k]=0;
return;
}
int m=l+r>>1;
if(L<=m)del(ld[k],l,m,L,R);
if(m<R)del(rd[k],m+1,r,L,R);
update(k);
}
int ask(int k,int l,int r,int sum){
if(l==r)return l;
int mid=l+r>>1;
if(sum<=d[ld[k]])return ask(ld[k],l,mid,sum);
return ask(rd[k],mid+1,r,sum-d[ld[k]]);
}
int main(){
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%d",&ol[i].c);
if(ol[i].c==1){
scanf("%d",&ol[i].a);
mp[++mx]=ol[i].a;
}else
if(ol[i].c==3||ol[i].c==4){
scanf("%d%d",&ol[i].a,&ol[i].b);
mp[++mx]=ol[i].b;
}else
if(ol[i].c==2||ol[i].c==5||ol[i].c==6)scanf("%d%d",&ol[i].a,&ol[i].b);else
scanf("%d",&ol[i].a);
}
sort(mp+1,mp+mx+1);
mx=unique(mp+1,mp+mx+1)-mp-1;
for(int i=1;i<=mx;i++)p[mp[i]]=i;
for(int i=1;i<=m;i++){
if(ol[i].c==1){
++n;
fa[n]=n;
add(root[n],1,mx,p[ol[i].a],1);
}else
if(ol[i].c==2){
u=dad(ol[i].a),v=dad(ol[i].b);
if(u!=v){
fa[v]=u;
root[u]=merge(root[u],root[v],1,mx);
}
}else
if(ol[i].c==3){
u=dad(ol[i].a),x=p[ol[i].b];
if(x==1)continue;
cnt=query(root[u],1,mx,1,x-1);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,1,x-1);
}else
if(ol[i].c==4){
u=dad(ol[i].a),x=p[ol[i].b];
if(x==mx)continue;
cnt=query(root[u],1,mx,x+1,mx);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,x+1,mx);
}else
if(ol[i].c==5){
u=dad(ol[i].a);
printf("%d\n",mp[ask(root[u],1,mx,ol[i].b)]);
}else
if(ol[i].c==6){
u=dad(ol[i].a),v=dad(ol[i].b);
printf("%d\n",lg[root[u]]>lg[root[v]]?1:0);
}else
if(ol[i].c==7)
printf("%d\n",d[root[dad(ol[i].a)]]);
}
return 0;
}
然而我看到最快的人的耗时才1800+ms,我却有18000+ms,就感觉不爽,于是加了读入优化(比平常的读优更高级)。
C++ Code:
#include<cstdio>
#include<algorithm>
#include<map>
#include<cmath>
#include<cctype>
#define N 6000000
using namespace std;
int m,n=0,mx=0,node=0,u,v,x,cnt;
int fa[400001],root[400001],ld[N],rd[N],mp[400001],d[N];
double lg[N]={0};
char buf[30000020];
int bufpos;
inline void init(){
buf[fread(buf,1,30000020,stdin)]='\0';
bufpos=0;
}
inline int readint(){
int p=0;
for(;!isdigit(buf[bufpos]);bufpos++);
for(;isdigit(buf[bufpos]);bufpos++)
p=p*10+buf[bufpos]-'0';
return p;
}
struct operation{
int c,a,b;
}ol[400001];
map<int,int>p;
inline void update(int k){
d[k]=d[ld[k]]+d[rd[k]];
lg[k]=lg[ld[k]]+lg[rd[k]];
}
int dad(int x){return fa[x]==x?x:(fa[x]=dad(fa[x]));}
void add(int& k,int l,int r,int x,int num){
if(!k)k=++node;
if(l==r){
d[k]+=num;
lg[k]+=num*log(mp[x]);
return;
}
int mid=l+r>>1;
if(x<=mid)add(ld[k],l,mid,x,num);else
add(rd[k],mid+1,r,x,num);
update(k);
}
inline int merge(int u,int v,int l,int r){
if(!u||!v)return u|v;
if(l==r){
d[u]+=d[v];
lg[u]+=lg[v];
return u;
}
int mid=l+r>>1;
ld[u]=merge(ld[u],ld[v],l,mid);
rd[u]=merge(rd[u],rd[v],mid+1,r);
update(u);
return u;
}
int query(int k,int l,int r,int L,int R){
if(L<=l&&r<=R)return d[k];
int m=l+r>>1,ans=0;
if(L<=m)ans+=query(ld[k],l,m,L,R);
if(m<R)ans+=query(rd[k],m+1,r,L,R);
return ans;
}
void del(int k,int l,int r,int L,int R){
if(!d[k])return;
if(l==r){
d[k]=0;
lg[k]=0;
return;
}
int m=l+r>>1;
if(L<=m)del(ld[k],l,m,L,R);
if(m<R)del(rd[k],m+1,r,L,R);
update(k);
}
int ask(int k,int l,int r,int sum){
if(l==r)return l;
int mid=l+r>>1;
if(sum<=d[ld[k]])return ask(ld[k],l,mid,sum);
return ask(rd[k],mid+1,r,sum-d[ld[k]]);
}
int main(){
init();
m=readint();
for(int i=1;i<=m;i++){
ol[i].c=readint();
if(ol[i].c==1)mp[++mx]=ol[i].a=readint();else
if(ol[i].c==3||ol[i].c==4){
ol[i].a=readint();
mp[++mx]=ol[i].b=readint();
}else
if(ol[i].c==2||ol[i].c==5||ol[i].c==6)ol[i].a=readint(),ol[i].b=readint();else
ol[i].a=readint();
}
sort(mp+1,mp+mx+1);
mx=unique(mp+1,mp+mx+1)-mp-1;
for(int i=1;i<=mx;i++)p[mp[i]]=i;
for(int i=1;i<=m;i++){
if(ol[i].c==1){
++n;
fa[n]=n;
add(root[n],1,mx,p[ol[i].a],1);
}else
if(ol[i].c==2){
u=dad(ol[i].a),v=dad(ol[i].b);
if(u!=v){
fa[v]=u;
root[u]=merge(root[u],root[v],1,mx);
}
}else
if(ol[i].c==3){
u=dad(ol[i].a),x=p[ol[i].b];
if(x==1)continue;
cnt=query(root[u],1,mx,1,x-1);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,1,x-1);
}else
if(ol[i].c==4){
u=dad(ol[i].a),x=p[ol[i].b];
if(x==mx)continue;
cnt=query(root[u],1,mx,x+1,mx);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,x+1,mx);
}else
if(ol[i].c==5){
u=dad(ol[i].a);
printf("%d\n",mp[ask(root[u],1,mx,ol[i].b)]);
}else
if(ol[i].c==6){
u=dad(ol[i].a),v=dad(ol[i].b);
printf("%d\n",lg[root[u]]>lg[root[v]]?1:0);
}else
if(ol[i].c==7)
printf("%d\n",d[root[dad(ol[i].a)]]);
}
return 0;
}
(上面是读优)快了3000ms左右,然而还是好慢。
不过能说明在数据较大的题目里,读入优化能大大提高代码效率。
那个1800+ms的是怎么做到的?orz
后来我又想到map可能会拖慢时间,于是map改成了直接用lower_bound()找。又快了2500ms左右。
C++ Code:
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cctype>
#define N 6000000
using namespace std;
int m,n=0,mx=0,node=0,u,v,x,cnt;
int fa[400001],root[400001],ld[N],rd[N],mp[400001],d[N];
double lg[N]={0};
char buf[30000020];
int bufpos;
inline int fnd(int x){
return lower_bound(mp+1,mp+1+mx,x)-mp;
}
inline void init(){
buf[fread(buf,1,30000020,stdin)]='\0';
bufpos=0;
}
inline int readint(){
int p=0;
for(;!isdigit(buf[bufpos]);bufpos++);
for(;isdigit(buf[bufpos]);bufpos++)
p=p*10+buf[bufpos]-'0';
return p;
}
struct operation{
int c,a,b;
}ol[400001];
inline void update(int k){
d[k]=d[ld[k]]+d[rd[k]];
lg[k]=lg[ld[k]]+lg[rd[k]];
}
int dad(int x){return fa[x]==x?x:(fa[x]=dad(fa[x]));}
void add(int& k,int l,int r,int x,int num){
if(!k)k=++node;
if(l==r){
d[k]+=num;
lg[k]+=num*log(mp[x]);
return;
}
int mid=l+r>>1;
if(x<=mid)add(ld[k],l,mid,x,num);else
add(rd[k],mid+1,r,x,num);
update(k);
}
inline int merge(int u,int v,int l,int r){
if(!u||!v)return u|v;
if(l==r){
d[u]+=d[v];
lg[u]+=lg[v];
return u;
}
int mid=l+r>>1;
ld[u]=merge(ld[u],ld[v],l,mid);
rd[u]=merge(rd[u],rd[v],mid+1,r);
update(u);
return u;
}
int query(int k,int l,int r,int L,int R){
if(L<=l&&r<=R)return d[k];
int m=l+r>>1,ans=0;
if(L<=m)ans+=query(ld[k],l,m,L,R);
if(m<R)ans+=query(rd[k],m+1,r,L,R);
return ans;
}
void del(int k,int l,int r,int L,int R){
if(!d[k])return;
if(l==r){
d[k]=0;
lg[k]=0;
return;
}
int m=l+r>>1;
if(L<=m)del(ld[k],l,m,L,R);
if(m<R)del(rd[k],m+1,r,L,R);
update(k);
}
int ask(int k,int l,int r,int sum){
if(l==r)return l;
int mid=l+r>>1;
if(sum<=d[ld[k]])return ask(ld[k],l,mid,sum);
return ask(rd[k],mid+1,r,sum-d[ld[k]]);
}
int main(){
init();
m=readint();
for(int i=1;i<=m;i++){
ol[i].c=readint();
if(ol[i].c==1)mp[++mx]=ol[i].a=readint();else
if(ol[i].c==3||ol[i].c==4){
ol[i].a=readint();
mp[++mx]=ol[i].b=readint();
}else
if(ol[i].c==2||ol[i].c==5||ol[i].c==6)ol[i].a=readint(),ol[i].b=readint();else
ol[i].a=readint();
}
sort(mp+1,mp+mx+1);
mx=unique(mp+1,mp+mx+1)-mp-1;
for(int i=1;i<=m;i++){
if(ol[i].c==1){
++n;
fa[n]=n;
add(root[n],1,mx,fnd(ol[i].a),1);
}else
if(ol[i].c==2){
u=dad(ol[i].a),v=dad(ol[i].b);
if(u!=v){
fa[v]=u;
root[u]=merge(root[u],root[v],1,mx);
}
}else
if(ol[i].c==3){
u=dad(ol[i].a),x=fnd(ol[i].b);
if(x==1)continue;
cnt=query(root[u],1,mx,1,x-1);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,1,x-1);
}else
if(ol[i].c==4){
u=dad(ol[i].a),x=fnd(ol[i].b);
if(x==mx)continue;
cnt=query(root[u],1,mx,x+1,mx);
add(root[u],1,mx,x,cnt);
del(root[u],1,mx,x+1,mx);
}else
if(ol[i].c==5){
u=dad(ol[i].a);
printf("%d\n",mp[ask(root[u],1,mx,ol[i].b)]);
}else
if(ol[i].c==6){
u=dad(ol[i].a),v=dad(ol[i].b);
printf("%d\n",lg[root[u]]>lg[root[v]]?1:0);
}else
if(ol[i].c==7)
printf("%d\n",d[root[dad(ol[i].a)]]);
}
return 0;
}