Submission #1177882

#TimeUsernameProblemLanguageResultExecution timeMemory
1177882sasde다리 (APIO19_bridges)C++17
13 / 100
613 ms3524 KiB
#include<bits/stdc++.h>
#define task "strdel"
#define ii pair<int,int>
#define iii pair<int,ii>
#define iv pair<ii,ii>
#define se second
#define fi first
#define ffi fi.fi
#define sfi se.fi
#define sse se.se
#define fse fi.se
#define lt(i, c, d) for(int i = c; i <= d; ++i)
#define fl(i, c, d) for(int i = d; i >= c; --i)
#define pb push_back
#define emb emplace_back
#define emf emplace_front
#define em emplace
using namespace std;
const int N=5e4+1,sq=450;
mt19937 rd(chrono::steady_clock::now().time_since_epoch().count());
int Rand(int u,int v){
 return u+rd()%(v-u+1);
}
struct zz{
  int n;
stack<ii>res;
vector<int>r;
void build(int _n){
n=_n;
r=vector<int>(n+5,-1);
}
int acs(int u){
  return r[u]<0?u:acs(r[u]);
}
void join(int u,int v,int i){
  u=acs(u);
  v=acs(v);
  if(u!=v){
    if(r[u]>r[v])swap(u,v);
    if(i==1)
    res.push({v,r[v]});
    r[u]+=r[v];
    r[v]=u;
  }
}
void rollback(){
  while(res.size()){
    ii x=res.top();
    res.pop();
    r[r[x.fi]]-=x.se;
    r[x.fi]=x.se;
  }
}
}s;
int m,n,q,ans[N];
struct pt{
  int u,v,w;
}b[N*2];
bool k[N];
int k1[N];
struct pt1{
  int t,u,w;
}c[N*2];
void solve(){
  cin >> n >> m;
  for(int i=1;i<=m;++i)cin >> b[i].u >> b[i].v >> b[i].w;
  cin >> q;
  for(int i=1;i<=q;++i)cin >> c[i].t >> c[i].u >> c[i].w;
  for(int st=1;st<=q;st+=sq){
    int en=min(q,st+sq-1);
    vector<int>jo,nojo,res;
    for(int i=st;i<=en;++i){
      if(c[i].t==1)k[c[i].u]=1;
      else res.emb(i);
    }
    for(int i=1;i<=m;++i){
      if(!k[i])nojo.emb(i);
      else jo.emb(i);
    }
    sort(res.begin(),res.end(),[](const int &x,const int &y){
      return c[x].w>c[y].w;
    });
    sort(nojo.begin(),nojo.end(),[](const int &x,const int &y){
      return b[x].w>b[y].w;
    });
    int j=0;s.build(n);
    for(auto x:res){
        // if(x==1)cout <<c[x].w<<b[nojo[j]].w<<" "<<'\n';
      while(j<nojo.size()&&b[nojo[j]].w>=c[x].w){
        // cout <<j<<" ";
        s.join(b[nojo[j]].u,b[nojo[j]].v,0);
        ++j;
      }
      for(int i=st;i<x;++i){
        if(c[i].t==1)k1[c[i].u]=i;
      }
      for(auto y:jo){
        if(k1[y]){
          if(c[k1[y]].w>=c[x].w)
            s.join(b[y].u,b[y].v,1);
        }
        else{
          if(b[y].w>=c[x].w)
            s.join(b[y].u,b[y].v,1);
        }
      }
      ans[x]=-s.r[s.acs(c[x].u)];
      for(int i=st;i<x;++i){
        if(c[i].t==1)k1[c[i].u]=0;
      }
      s.rollback();

    }

    for(int i=st;i<=en;++i){
      if(c[i].t==1){
        k[c[i].u]=0;
        b[c[i].u].w=c[i].w;
      }
    }

  }
  for(int i=1;i<=q;++i)if(c[i].t==2)cout <<ans[i]<<'\n';
}
main()
{
  srand(time(0));
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    if(fopen(task".inp","r")){
      freopen(task".inp","r",stdin);
      freopen(task".out","w",stdout);
    }
    int t=1;
 //   cin >> t;
while(t--){
    solve();
}

}

Compilation message (stderr)

bridges.cpp:125:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  125 | main()
      | ^~~~
bridges.cpp: In function 'int main()':
bridges.cpp:132:14: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  132 |       freopen(task".inp","r",stdin);
      |       ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
bridges.cpp:133:14: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  133 |       freopen(task".out","w",stdout);
      |       ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...