update
This commit is contained in:
parent
ed231e6777
commit
1335cabe0a
@ -1,179 +1,182 @@
|
|||||||
#include<bits/stdc++.h>
|
// #include<bits/stdc++.h>
|
||||||
#define ll long long
|
// #define ll long long
|
||||||
using namespace std;
|
// using namespace std;
|
||||||
const int maxn=2e5+5;
|
// const int maxn=2e5+5;
|
||||||
struct node
|
// struct node
|
||||||
{
|
// {
|
||||||
ll mx,sum,t,sv,add,tag,len;
|
// ll mx,sum,t,sv,add,tag,len;
|
||||||
};
|
// };
|
||||||
node tr[maxn<<2];
|
// node tr[maxn<<2];
|
||||||
int n,q;
|
// int n,q;
|
||||||
ll d[maxn<<2];
|
// ll d[maxn<<2];
|
||||||
inline void mdf(int id,int k,ll w)
|
// inline void mdf(int id,int k,ll w)
|
||||||
{
|
// {
|
||||||
if(k==0)
|
// if(k==0)
|
||||||
{
|
// {
|
||||||
tr[id].mx+=w;
|
// tr[id].mx+=w;
|
||||||
tr[id].t+=w*w*tr[id].len+tr[id].sv*w+tr[id].sum*w;
|
// tr[id].t+=w*w*tr[id].len+tr[id].sv*w+tr[id].sum*w;
|
||||||
tr[id].sum+=w*tr[id].len;
|
// tr[id].sum+=w*tr[id].len;
|
||||||
tr[id].sv+=w*tr[id].len;
|
// tr[id].sv+=w*tr[id].len;
|
||||||
tr[id].add+=w;
|
// tr[id].add+=w;
|
||||||
}
|
// }
|
||||||
if(k==1)
|
// if(k==1)
|
||||||
{
|
// {
|
||||||
tr[id].mx=w,tr[id].t=w*w*tr[id].len,tr[id].sum=w*tr[id].len,tr[id].sv=w*tr[id].len;
|
// tr[id].mx=w,tr[id].t=w*w*tr[id].len,tr[id].sum=w*tr[id].len,tr[id].sv=w*tr[id].len;
|
||||||
tr[id].add=0,tr[id].tag=w;
|
// tr[id].add=0,tr[id].tag=w;
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
inline void pushdown(int id)
|
// inline void pushdown(int id)
|
||||||
{
|
// {
|
||||||
if(tr[id].tag)
|
// if(tr[id].tag)
|
||||||
{
|
// {
|
||||||
mdf(id<<1,1,tr[id].tag);
|
// mdf(id<<1,1,tr[id].tag);
|
||||||
mdf(id<<1|1,1,tr[id].tag);
|
// mdf(id<<1|1,1,tr[id].tag);
|
||||||
}
|
// }
|
||||||
if(tr[id].add)
|
// if(tr[id].add)
|
||||||
{
|
// {
|
||||||
mdf(id<<1,0,tr[id].add);
|
// mdf(id<<1,0,tr[id].add);
|
||||||
mdf(id<<1|1,0,tr[id].add);
|
// mdf(id<<1|1,0,tr[id].add);
|
||||||
}
|
// }
|
||||||
tr[id].tag=tr[id].add=0;
|
// tr[id].tag=tr[id].add=0;
|
||||||
}
|
// }
|
||||||
node calc(node nd,int id)
|
// node calc(node nd,int id)
|
||||||
{
|
// {
|
||||||
node ret=nd;
|
// node ret=nd;
|
||||||
if(tr[id].len==1)
|
// if(tr[id].len==1)
|
||||||
{
|
// {
|
||||||
ret.mx=max(ret.mx,tr[id].mx);
|
// ret.mx=max(ret.mx,tr[id].mx);
|
||||||
ret.sum+=tr[id].sum;
|
// ret.sum+=tr[id].sum;
|
||||||
ret.t+=tr[id].sum*ret.mx;
|
// ret.t+=tr[id].sum*ret.mx;
|
||||||
ret.sv+=ret.mx;
|
// ret.sv+=ret.mx;
|
||||||
ret.len+=tr[id].len;
|
// ret.len+=tr[id].len;
|
||||||
}
|
// }
|
||||||
else
|
// else
|
||||||
{
|
// {
|
||||||
pushdown(id);
|
// pushdown(id);
|
||||||
if(tr[id<<1].mx>=nd.mx)
|
// if(tr[id<<1].mx>=nd.mx)
|
||||||
{
|
// {
|
||||||
ret=calc(nd,id<<1);
|
// ret=calc(nd,id<<1);
|
||||||
ret.mx=max(ret.mx,tr[id<<1|1].mx);
|
// ret.mx=max(ret.mx,tr[id<<1|1].mx);
|
||||||
ret.sum+=tr[id<<1|1].sum;
|
// ret.sum+=tr[id<<1|1].sum;
|
||||||
ret.t+=tr[id].t-tr[id<<1].t;
|
// ret.t+=tr[id].t-tr[id<<1].t;
|
||||||
ret.sv+=tr[id].sv-tr[id<<1].sv;
|
// ret.sv+=tr[id].sv-tr[id<<1].sv;
|
||||||
ret.len+=tr[id<<1|1].len;
|
// ret.len+=tr[id<<1|1].len;
|
||||||
}
|
// }
|
||||||
else
|
// else
|
||||||
{
|
// {
|
||||||
nd.sum+=tr[id<<1].sum;
|
// nd.sum+=tr[id<<1].sum;
|
||||||
nd.t+=tr[id<<1].sum*nd.mx;
|
// nd.t+=tr[id<<1].sum*nd.mx;
|
||||||
nd.sv+=tr[id<<1].len*nd.mx;
|
// nd.sv+=tr[id<<1].len*nd.mx;
|
||||||
nd.len+=tr[id<<1].len;
|
// nd.len+=tr[id<<1].len;
|
||||||
ret=calc(nd,id<<1|1);
|
// ret=calc(nd,id<<1|1);
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
return ret;
|
// return ret;
|
||||||
}
|
// }
|
||||||
inline void pushup(int id)
|
// inline void pushup(int id)
|
||||||
{
|
// {
|
||||||
tr[id]=calc(tr[id<<1],id<<1|1);
|
// tr[id]=calc(tr[id<<1],id<<1|1);
|
||||||
tr[id].tag=tr[id].add=0;
|
// tr[id].tag=tr[id].add=0;
|
||||||
}
|
// }
|
||||||
void build(int id,int l,int r)
|
// void build(int id,int l,int r)
|
||||||
{
|
// {
|
||||||
if(l==r)
|
// if(l==r)
|
||||||
{
|
// {
|
||||||
tr[id]=(node){d[l],d[l],d[l]*d[l],d[l],0,0,1};
|
// tr[id]=(node){d[l],d[l],d[l]*d[l],d[l],0,0,1};
|
||||||
return;
|
// return;
|
||||||
}
|
// }
|
||||||
int mid=(l+r)>>1;
|
// int mid=(l+r)>>1;
|
||||||
build(id<<1,l,mid);
|
// build(id<<1,l,mid);
|
||||||
build(id<<1|1,mid+1,r);
|
// build(id<<1|1,mid+1,r);
|
||||||
pushup(id);
|
// pushup(id);
|
||||||
}
|
// }
|
||||||
void update(int id,int l,int r,int x,int y,int k,ll w)
|
// void update(int id,int l,int r,int x,int y,int k,ll w)
|
||||||
{
|
// {
|
||||||
if(l>=x && r<=y)
|
// if(l>=x && r<=y)
|
||||||
{
|
// {
|
||||||
mdf(id,k,w);
|
// mdf(id,k,w);
|
||||||
return;
|
// return;
|
||||||
}
|
// }
|
||||||
int mid=(l+r)>>1;
|
// int mid=(l+r)>>1;
|
||||||
pushdown(id);
|
// pushdown(id);
|
||||||
if(x<=mid) update(id<<1,l,mid,x,y,k,w);
|
// if(x<=mid) update(id<<1,l,mid,x,y,k,w);
|
||||||
if(y>mid) update(id<<1|1,mid+1,r,x,y,k,w);
|
// if(y>mid) update(id<<1|1,mid+1,r,x,y,k,w);
|
||||||
pushup(id);
|
// pushup(id);
|
||||||
}
|
// }
|
||||||
int num[maxn],cnt,L[maxn],R[maxn];
|
// int num[maxn],cnt,L[maxn],R[maxn];
|
||||||
void get_num(int id,int l,int r,int x,int y)
|
// void get_num(int id,int l,int r,int x,int y)
|
||||||
{
|
// {
|
||||||
if(l>=x && r<=y)
|
// if(l>=x && r<=y)
|
||||||
{
|
// {
|
||||||
cnt++;
|
// cnt++;
|
||||||
num[cnt]=id,L[cnt]=l,R[cnt]=r;
|
// num[cnt]=id,L[cnt]=l,R[cnt]=r;
|
||||||
return;
|
// return;
|
||||||
}
|
// }
|
||||||
int mid=(l+r)>>1;
|
// int mid=(l+r)>>1;
|
||||||
pushdown(id);
|
// pushdown(id);
|
||||||
if(x<=mid) get_num(id<<1,l,mid,x,y);
|
// if(x<=mid) get_num(id<<1,l,mid,x,y);
|
||||||
if(y>mid) get_num(id<<1|1,mid+1,r,x,y);
|
// if(y>mid) get_num(id<<1|1,mid+1,r,x,y);
|
||||||
}
|
// }
|
||||||
int get_pos(int id,int l,int r,node nd,ll t)
|
// int get_pos(int id,int l,int r,node nd,ll t)
|
||||||
{
|
// {
|
||||||
if(l==r) return l;
|
// if(l==r) return l;
|
||||||
int mid=(l+r)>>1;
|
// int mid=(l+r)>>1;
|
||||||
pushdown(id);
|
// pushdown(id);
|
||||||
node dt=calc(nd,id<<1);
|
// node dt=calc(nd,id<<1);
|
||||||
if(dt.t<=t) return get_pos(id<<1|1,mid+1,r,dt,t);
|
// if(dt.t<=t) return get_pos(id<<1|1,mid+1,r,dt,t);
|
||||||
else return get_pos(id<<1,l,mid,nd,t);
|
// else return get_pos(id<<1,l,mid,nd,t);
|
||||||
}
|
// }
|
||||||
int main()
|
// int main()
|
||||||
{
|
// {
|
||||||
//system("fc training.out training6.ans /W");
|
// //system("fc training.out training6.ans /W");
|
||||||
//return 0;
|
// //return 0;
|
||||||
ios::sync_with_stdio(0);
|
// ios::sync_with_stdio(0);
|
||||||
cin.tie(0);cout.tie(0);
|
// cin.tie(0);cout.tie(0);
|
||||||
//freopen("training6.in","r",stdin);
|
// //freopen("training6.in","r",stdin);
|
||||||
//freopen("training.out","w",stdout);
|
// //freopen("training.out","w",stdout);
|
||||||
int fl;
|
// int fl;
|
||||||
node nd,dt;
|
// node nd,dt;
|
||||||
char opt;
|
// char opt;
|
||||||
ll s,t,w;
|
// ll s,t,w;
|
||||||
cin>>n>>q;
|
// cin>>n>>q;
|
||||||
for(int i=0;i<n;i++) cin>>d[i];
|
// for(int i=0;i<n;i++) cin>>d[i];
|
||||||
build(1,0,n-1);
|
// build(1,0,n-1);
|
||||||
while(q--)
|
// while(q--)
|
||||||
{
|
// {
|
||||||
cin>>opt>>s>>t;
|
// cin>>opt>>s>>t;
|
||||||
if(opt=='P')
|
// if(opt=='P')
|
||||||
{
|
// {
|
||||||
cin>>w;
|
// cin>>w;
|
||||||
if(s<=t) update(1,0,n-1,s,t,0,w);
|
// if(s<=t) update(1,0,n-1,s,t,0,w);
|
||||||
else update(1,0,n-1,0,t,0,w),update(1,0,n-1,s,n-1,0,w);
|
// else update(1,0,n-1,0,t,0,w),update(1,0,n-1,s,n-1,0,w);
|
||||||
}
|
// }
|
||||||
if(opt=='R')
|
// if(opt=='R')
|
||||||
{
|
// {
|
||||||
cin>>w;
|
// cin>>w;
|
||||||
if(s<=t) update(1,0,n-1,s,t,1,w);
|
// if(s<=t) update(1,0,n-1,s,t,1,w);
|
||||||
else update(1,0,n-1,0,t,1,w),update(1,0,n-1,s,n-1,1,w);
|
// else update(1,0,n-1,0,t,1,w),update(1,0,n-1,s,n-1,1,w);
|
||||||
}
|
// }
|
||||||
if(opt=='Q')
|
// if(opt=='Q')
|
||||||
{
|
// {
|
||||||
fl=0,cnt=0,nd=(node){0,0,0,0,0,0,0};
|
// fl=0,cnt=0,nd=(node){0,0,0,0,0,0,0};
|
||||||
get_num(1,0,n-1,s,n-1);
|
// get_num(1,0,n-1,s,n-1);
|
||||||
get_num(1,0,n-1,0,n-1);
|
// get_num(1,0,n-1,0,n-1);
|
||||||
for(int i=1;i<=cnt;i++)
|
// for(int i=1;i<=cnt;i++)
|
||||||
{
|
// {
|
||||||
dt=calc(nd,num[i]);
|
// dt=calc(nd,num[i]);
|
||||||
if(dt.t>t)
|
// if(dt.t>t)
|
||||||
{
|
// {
|
||||||
fl=1;
|
// fl=1;
|
||||||
cout<<get_pos(num[i],L[i],R[i],nd,t)<<'\n';
|
// cout<<get_pos(num[i],L[i],R[i],nd,t)<<'\n';
|
||||||
break;
|
// break;
|
||||||
}
|
// }
|
||||||
nd=dt;
|
// nd=dt;
|
||||||
}
|
// }
|
||||||
if(fl==0) cout<<get_pos(1,0,n-1,nd,(t-nd.t)%(tr[1].sum*tr[1].mx)+nd.t)<<'\n';
|
// if(fl==0) cout<<get_pos(1,0,n-1,nd,(t-nd.t)%(tr[1].sum*tr[1].mx)+nd.t)<<'\n';
|
||||||
}
|
// }
|
||||||
}
|
// }
|
||||||
return 0;
|
// return 0;
|
||||||
|
// }
|
||||||
|
int main(){
|
||||||
|
|
||||||
}
|
}
|
||||||
@ -1,202 +1,3 @@
|
|||||||
#include<bits/stdc++.h>
|
|
||||||
using namespace std;
|
|
||||||
typedef long long ll;
|
|
||||||
const int N=200001;
|
|
||||||
int n,q;
|
|
||||||
int a[N];
|
|
||||||
struct node{
|
|
||||||
int mx;
|
|
||||||
ll sm,smx,tt;
|
|
||||||
int tg1,tg2;
|
|
||||||
}t[N*4];
|
|
||||||
#define ls (p<<1)
|
|
||||||
#define rs (p<<1|1)
|
|
||||||
#define mid (l+r>>1)
|
|
||||||
void build(int p,int l,int r){
|
|
||||||
for(int i=l;i<=r;i++){
|
|
||||||
t[p].mx=max(t[p].mx,a[i]);
|
|
||||||
t[p].sm+=a[i];
|
|
||||||
t[p].smx+=t[p].mx;
|
|
||||||
t[p].tt+=1ll*t[p].mx*a[i];
|
|
||||||
}
|
|
||||||
if(l==r)return;
|
|
||||||
build(ls,l,mid);
|
|
||||||
build(rs,mid+1,r);
|
|
||||||
}
|
|
||||||
void pushdown(int p,int l,int r){
|
|
||||||
int tg;
|
|
||||||
if(t[p].tg1){
|
|
||||||
tg=t[p].tg1;
|
|
||||||
t[p].tg1=0;
|
|
||||||
t[ls].mx=t[rs].mx=tg;
|
|
||||||
t[ls].sm=1ll*(mid-l+1)*tg,t[rs].sm=1ll*(r-mid)*tg;
|
|
||||||
t[ls].smx=t[ls].sm,t[rs].smx=t[rs].sm;
|
|
||||||
t[ls].tt=t[ls].sm*tg,t[rs].tt=t[rs].sm*tg;
|
|
||||||
t[ls].tg1=t[rs].tg1=tg;
|
|
||||||
t[ls].tg2=t[rs].tg2=0;
|
|
||||||
}
|
|
||||||
if(t[p].tg2){
|
|
||||||
tg=t[p].tg2;
|
|
||||||
t[p].tg2=0;
|
|
||||||
t[ls].mx+=tg,t[rs].mx+=tg;
|
|
||||||
t[ls].tt+=tg*(1ll*tg*(mid-l+1)+t[ls].sm+t[ls].smx),t[rs].tt+=tg*(1ll*tg*(r-mid)+t[rs].sm+t[rs].smx);
|
|
||||||
t[ls].sm+=1ll*(mid-l+1)*tg,t[rs].sm+=1ll*(r-mid)*tg;
|
|
||||||
t[ls].smx+=1ll*(mid-l+1)*tg,t[rs].smx+=1ll*(r-mid)*tg;
|
|
||||||
t[ls].tg2+=tg,t[rs].tg2+=tg;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
int qmx;
|
|
||||||
ll qsmx,qtt;
|
|
||||||
void query(int p,int l,int r){
|
|
||||||
if(l==r){
|
|
||||||
if(qmx>t[p].mx)qsmx+=qmx,qtt+=qmx*t[p].sm;
|
|
||||||
else qsmx+=t[p].mx,qtt+=t[p].tt,qmx=t[p].mx;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
if(qmx>t[ls].mx){
|
|
||||||
qsmx+=1ll*qmx*(mid-l+1);
|
|
||||||
qtt+=qmx*t[ls].sm;
|
|
||||||
query(rs,mid+1,r);
|
|
||||||
}
|
|
||||||
else{
|
|
||||||
qsmx+=t[p].smx-t[ls].smx;
|
|
||||||
qtt+=t[p].tt-t[ls].tt;
|
|
||||||
query(ls,l,mid);
|
|
||||||
qmx=t[p].mx;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
void update(int p,int l,int r){
|
|
||||||
t[p].mx=max(t[ls].mx,t[rs].mx);
|
|
||||||
t[p].sm=t[ls].sm+t[rs].sm;
|
|
||||||
qmx=t[ls].mx;
|
|
||||||
qsmx=t[ls].smx;
|
|
||||||
qtt=t[ls].tt;
|
|
||||||
query(rs,mid+1,r);
|
|
||||||
t[p].smx=qsmx;
|
|
||||||
t[p].tt=qtt;
|
|
||||||
}
|
|
||||||
void modify1(int p,int l,int r,int x,int y,int d){
|
|
||||||
if(x<=l&&r<=y){
|
|
||||||
t[p].mx=d;
|
|
||||||
t[p].sm=1ll*(r-l+1)*d;
|
|
||||||
t[p].smx=t[p].sm;
|
|
||||||
t[p].tt=t[p].sm*d;
|
|
||||||
t[p].tg1=d;
|
|
||||||
t[p].tg2=0;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
if(x<=mid)modify1(ls,l,mid,x,y,d);
|
|
||||||
if(mid<y)modify1(rs,mid+1,r,x,y,d);
|
|
||||||
update(p,l,r);
|
|
||||||
}
|
|
||||||
void modify2(int p,int l,int r,int x,int y,int d){
|
|
||||||
if(x<=l&&r<=y){
|
|
||||||
t[p].mx+=d;
|
|
||||||
t[p].tt+=d*(1ll*d*(r-l+1)+t[p].sm+t[p].smx);
|
|
||||||
t[p].sm+=1ll*(r-l+1)*d;
|
|
||||||
t[p].smx+=1ll*(r-l+1)*d;
|
|
||||||
t[p].tg2+=d;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
if(x<=mid)modify2(ls,l,mid,x,y,d);
|
|
||||||
if(mid<y)modify2(rs,mid+1,r,x,y,d);
|
|
||||||
update(p,l,r);
|
|
||||||
}
|
|
||||||
struct seg{
|
|
||||||
int p,l,r;
|
|
||||||
}b[100];
|
|
||||||
int m;
|
|
||||||
void getseg(int p,int l,int r,int x,int y){
|
|
||||||
if(x<=l&&r<=y){
|
|
||||||
b[++m]=(seg){p,l,r};
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
if(x<=mid)getseg(ls,l,mid,x,y);
|
|
||||||
if(mid<y)getseg(rs,mid+1,r,x,y);
|
|
||||||
}
|
|
||||||
int gettt(int p,int l,int r,ll tm){
|
|
||||||
if(l==r){
|
|
||||||
query(p,l,r);
|
|
||||||
if(qtt>tm)return l-1;
|
|
||||||
return l;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
int mx=qmx;
|
|
||||||
ll tt=qtt;
|
|
||||||
query(ls,l,mid);
|
|
||||||
if(qtt>tm){
|
|
||||||
qmx=mx,qtt=tt;
|
|
||||||
return gettt(ls,l,mid,tm);
|
|
||||||
}
|
|
||||||
return gettt(rs,mid+1,r,tm);
|
|
||||||
}
|
|
||||||
int getsm(int p,int l,int r,ll tm){
|
|
||||||
if(l==r){
|
|
||||||
if(t[p].sm>tm)return l-1;
|
|
||||||
return l;
|
|
||||||
}
|
|
||||||
pushdown(p,l,r);
|
|
||||||
if(t[ls].sm>tm)return getsm(ls,l,mid,tm);
|
|
||||||
return getsm(rs,mid+1,r,tm-t[ls].sm);
|
|
||||||
}
|
|
||||||
int main(){
|
int main(){
|
||||||
ios::sync_with_stdio(0);
|
|
||||||
cin.tie(0),cout.tie(0);
|
|
||||||
cin>>n>>q;
|
|
||||||
for(int i=0;i<n;i++)cin>>a[i];
|
|
||||||
build(1,0,n-1);
|
|
||||||
while(q--){
|
|
||||||
char op[2];
|
|
||||||
cin>>op;
|
|
||||||
if(op[0]=='R'){
|
|
||||||
int l,r,d;
|
|
||||||
cin>>l>>r>>d;
|
|
||||||
if(l<=r)modify1(1,0,n-1,l,r,d);
|
|
||||||
else modify1(1,0,n-1,l,n-1,d),modify1(1,0,n-1,0,r,d);
|
|
||||||
}
|
|
||||||
if(op[0]=='P'){
|
|
||||||
int l,r,d;
|
|
||||||
cin>>l>>r>>d;
|
|
||||||
if(l<=r)modify2(1,0,n-1,l,r,d);
|
|
||||||
else modify2(1,0,n-1,l,n-1,d),modify2(1,0,n-1,0,r,d);
|
|
||||||
}
|
|
||||||
if(op[0]=='Q'){
|
|
||||||
int s;
|
|
||||||
ll tm;
|
|
||||||
cin>>s>>tm;
|
|
||||||
m=0;
|
|
||||||
getseg(1,0,n-1,s,n-1);
|
|
||||||
if(s)getseg(1,0,n-1,0,s-1);
|
|
||||||
bool flag=0;
|
|
||||||
int mx=0;
|
|
||||||
ll tt=0;
|
|
||||||
for(int i=1;i<=m;i++){
|
|
||||||
qmx=mx,qtt=tt;
|
|
||||||
query(b[i].p,b[i].l,b[i].r);
|
|
||||||
if(qtt>tm){
|
|
||||||
qmx=mx,qtt=tt;
|
|
||||||
cout<<(gettt(b[i].p,b[i].l,b[i].r,tm)+1)%n<<'\n';
|
|
||||||
flag=1;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
mx=qmx,tt=qtt;
|
|
||||||
}
|
|
||||||
if(flag)continue;
|
|
||||||
tm-=qtt;
|
|
||||||
tm/=t[1].mx;
|
|
||||||
tm%=t[1].sm;
|
|
||||||
for(int i=1;i<=m;i++){
|
|
||||||
if(t[b[i].p].sm>tm){
|
|
||||||
cout<<(getsm(b[i].p,b[i].l,b[i].r,tm)+1)%n<<'\n';
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
tm-=t[b[i].p].sm;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
@ -1,9 +1,11 @@
|
|||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <queue>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
using ll = int64_t;
|
using ll = int64_t;
|
||||||
using vl = std::vector<ll>;
|
using vl = std::vector<ll>;
|
||||||
|
#define N(v)#v<<" :"<<(v)<<" "
|
||||||
auto &ci = std::cin;
|
auto &ci = std::cin;
|
||||||
auto &co = std::cout;
|
auto &co = std::cout;
|
||||||
|
|
||||||
@ -12,10 +14,10 @@ ll *sn = new ll[max_n];
|
|||||||
auto *nds = new vl[max_n];
|
auto *nds = new vl[max_n];
|
||||||
|
|
||||||
void dfs(ll ft, ll n){
|
void dfs(ll ft, ll n){
|
||||||
if(nds[n].size()<=1){
|
sn[n]=1;
|
||||||
|
if(nds[n].size()==1 && nds[n][0]==ft){
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
sn[n]=1;
|
|
||||||
for(ll i:nds[n]){
|
for(ll i:nds[n]){
|
||||||
if(i==ft){
|
if(i==ft){
|
||||||
continue;
|
continue;
|
||||||
@ -35,6 +37,43 @@ int main(){
|
|||||||
nds[a].push_back(b);
|
nds[a].push_back(b);
|
||||||
nds[b].push_back(a);
|
nds[b].push_back(a);
|
||||||
}
|
}
|
||||||
for(ll i{1};i<=n;i++)co<<sn[i]<<' ';
|
dfs(0, 1);
|
||||||
co<<'\n';
|
|
||||||
|
ll now{1};
|
||||||
|
ll nxt{}, nn{-1};
|
||||||
|
ll f{1};
|
||||||
|
for(ll i:nds[now]){
|
||||||
|
if(sn[i]>nn && i!=now){
|
||||||
|
nn = sn[i];
|
||||||
|
nxt = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
co<<"nxt: "<<nxt<<" now: "<<now<<"\n";
|
||||||
|
ll t{sn[1]-sn[nxt]};
|
||||||
|
ll b{sn[nxt]-1};
|
||||||
|
|
||||||
|
// ll fsn{sn[1]};
|
||||||
|
// co<<"t "<<t<<" b "<<b<<'\n';
|
||||||
|
ll ft{t}, fb{b};
|
||||||
|
while(t<b){
|
||||||
|
if(nds[now].size()==0)break;
|
||||||
|
f = now;
|
||||||
|
now = nxt;
|
||||||
|
nn = -1;
|
||||||
|
for(ll i{0}; i<nds[now].size(); i++){
|
||||||
|
for(ll i:nds[now]){
|
||||||
|
if(sn[i]>nn && i!=f){
|
||||||
|
nn = sn[i];
|
||||||
|
nxt = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ft=t, fb = b;
|
||||||
|
t+=(sn[now]-sn[nxt]);
|
||||||
|
b = sn[nxt]-1;
|
||||||
|
co<<"now: "<<now<<" nxt:"<<nxt<<" t: "<<t<<" b:"<<b<<'\n';
|
||||||
|
}
|
||||||
|
|
||||||
|
// co<<N(ft)<<N(fb)<<N(now)<<N(nxt)<<'\n';
|
||||||
|
|
||||||
}
|
}
|
||||||
Loading…
Reference in New Issue
Block a user