Submission #386868

#TimeUsernameProblemLanguageResultExecution timeMemory
386868uacoder123Factories (JOI14_factories)C++14
33 / 100
8068 ms229896 KiB
    #include "factories.h"
    #include <bits/stdc++.h>
    #include <ext/pb_ds/assoc_container.hpp>
    using namespace std;
    using namespace __gnu_pbds;
    #define F first
    #define S second
    #define FOR(i,a,b) for (auto i = (a); i <= (b); ++i)
    #define NFOR(i,a,b) for(auto i = (a); i >= (b); --i)
    #define all(x) (x).begin(), (x).end()
    #define sz(x) lli(x.size())
    #define mp(i,a) make_pair(i,a)
    #define pb(a) push_back(a)
    #define bit(x,b) (x&(1LL<<b))
    typedef long long int lli;
    typedef pair <lli,lli> ii;
    typedef pair <ii,lli> iii;
    typedef vector <lli> vi;
    typedef tree<lli,null_type,less<lli>,rb_tree_tag,tree_order_statistics_node_update> ordered_set;
    int n;
    vector<ii> al[500001];
    lli dp[500001],par[500001][25],par1[500001],lev[500001],din[500001],dout[500001],t=0,sz[500001],vis[500001],vis1[500001],pre[500001],r;
    void dfs(int node)
    {
      sz[node]=1;
      vis1[node]=1;
      for(int i=0;i<al[node].size();++i)
      {
        if(vis1[al[node][i].F]+vis[al[node][i].F]==0)
        {
          dfs(al[node][i].F);
          sz[node]+=sz[al[node][i].F];
        }
      }
    }
    int tell(int node,int p,int to)
    {
        for(int i=0;i<al[node].size();++i)
        {
          if(al[node][i].F!=p&&sz[al[node][i].F]>to/2&&vis[al[node][i].F]==0)
            return tell(al[node][i].F,node,to);
        }
        return node;
    }
    void dfs1(int node,int p,int cen)
    {
      pre[node]=cen;
      for(int i=0;i<al[node].size();++i)
        if(al[node][i].F!=p&&vis[al[node][i].F]==0)
          dfs1(al[node][i].F,node,cen);
    }
    void dfs2(int node,int p,lli l)
    {
      par[node][0]=p;
      for(int i=1;i<25;++i)
        par[node][i]=par[par[node][i-1]][i-1];
      lev[node]=l;
      din[node]=t++;
      for(int i=0;i<al[node].size();++i)
      {
        if(al[node][i].F!=p)
          dfs2(al[node][i].F,node,l+al[node][i].S);
      }
      dout[node]=t++;
    }
    bool isa(int f,int s)
    {
      return (din[f]<=din[s])&&(dout[f]>=dout[s]);
    }
    int lca(int f,int s)
    {
      if(isa(f,s))
        return f;
      if(isa(s,f))
        return s;
      for(int i=24;i>=0;--i)
        if(!isa(par[f][i],s))
          f=par[f][i];
      return par[f][0];
    }
    void Init(int N, int A[], int B[], int D[]) 
    {
      
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
      n=N;
      for(int i=0;i<n-1;++i)
      {
        al[A[i]].pb(mp(B[i],D[i]));
        al[B[i]].pb(mp(A[i],D[i]));
      }
      dfs2(0,0,0);
      int c=0;
      while(c!=n)
      {
        for(int i=0;i<n;++i)
        {
          if(vis[i]+vis1[i]==0)
          {
            dfs(i);
            int cen=tell(i,i,sz[i]);
            par1[cen]=pre[i];
            dfs1(i,i,cen);
            c++;
            vis[cen]=1;
            if(c==1)
              r=cen;
          }
        }
        memset(vis1,0,sizeof(vis1));
      }
      for(int i=0;i<n;++i)
        dp[i]=1000000000ll*5000000;
    }
     
    long long Query(int S, int X[], int T, int Y[]) 
    {
      vi v;
      for(lli i=0;i<S;++i)
      {
      lli cur=X[i];
          while(1)
          {
            lli l=lca(X[i],cur);
            dp[cur]=min(dp[cur],lev[X[i]]+lev[cur]-2*lev[l]);
            v.pb(cur);
            if(cur==r)
              break;
            cur=par1[cur];
          }
        }
        lli ans=1000000000ll*5000000;
        for(lli i=0;i<T;++i)
        {
     
         lli cur=Y[i];
          while(1)
          {
            lli l=lca(Y[i],cur);
            ans=min(ans,dp[cur]+lev[Y[i]]+lev[cur]-2*lev[l]);
            if(cur==r)
              break;
            cur=par1[cur];
          }
        }
        while(v.size())
        {
          dp[v.back()]=1000000000ll*5000000;
          v.pop_back();
        }
      return ans;
    }

Compilation message (stderr)

factories.cpp: In function 'void dfs(int)':
factories.cpp:27:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   27 |       for(int i=0;i<al[node].size();++i)
      |                   ~^~~~~~~~~~~~~~~~
factories.cpp: In function 'int tell(int, int, int)':
factories.cpp:38:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |         for(int i=0;i<al[node].size();++i)
      |                     ~^~~~~~~~~~~~~~~~
factories.cpp: In function 'void dfs1(int, int, int)':
factories.cpp:48:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |       for(int i=0;i<al[node].size();++i)
      |                   ~^~~~~~~~~~~~~~~~
factories.cpp: In function 'void dfs2(int, int, lli)':
factories.cpp:59:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |       for(int i=0;i<al[node].size();++i)
      |                   ~^~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...