Submission #864502

# Submission time Handle Problem Language Result Execution time Memory
864502 2023-10-23T05:49:59 Z HuyQuang_re_Zero Islands (IOI08_islands) C++14
90 / 100
828 ms 131072 KB
#include <bits/stdc++.h>
#define ll long long
#define db long double
#define N 1000005
#define II pair <int,int>
#define III pair <ll,II>
#define IV pair <vector <int>,vector <int> >
#define fst first
#define snd second
#define BIT(x,i) ((x>>i)&1)
#define Log(x) (31-__builtin_clz((int)x))
#define LogLL(x) (63-__builtin_clzll((ll)x))
#define pi acos(-1)
#define to_radian(x) (x*pi/180.0)
#define to_degree(x) (x*180.0/pi)
using namespace std;
vector <int> a[N];
int tr[N],trc[N],nxt[N],w[N];
bool incycle[N],visited[N];
ll n,i,u,v,k,ncycle,sum,res;
vector <ll> dis,c;
const ll oo=round(1e18);

void dfs(int u)
{
    visited[u]=1;
    auto consider=[&](int v)
    {
        if(visited[v]==0) dfs(v);
    };
    for(int adj:a[u]) consider(adj);
    consider(nxt[u]);
}

void Find_cycle(int u,int x)
{
    auto consider=[&](int v,int From)
    {
        if(From==x) return;
        int k=w[From];
        if(tr[v]==0)
        {
            tr[v]=u; trc[v]=k;
            Find_cycle(v,From);
        }
        else if(ncycle==0)
        {
            sum=0;
            while(u!=v)
            {
                ncycle++;
                c.push_back(u); incycle[u]=1;
                dis.push_back(sum); sum+=trc[u];
                u=tr[u];
            }
            ncycle++;
            c.push_back(u); incycle[u]=1;
            dis.push_back(sum); sum+=k;
            return ;
        }
    };
    for(int adj:a[u])
    {
        consider(adj,adj);
        if(ncycle>0) return ;
    }
    consider(nxt[u],u);
}


/////////////////////////////////////////////////////////////////////////////
ll cal(int u,int p,ll &ma)
{
    ll res=0,ma0=0,ma1=0;
    auto consider=[&](int adj,int From)
    {
        int v=adj;
        if(v!=p && incycle[v]==0)
        {
            ll k=cal(v,u,ma)+w[From];
            res=max(res,k);
            if(ma0<k) ma0=k;
            if(ma1<ma0) swap(ma1,ma0);
        }
    };
    for(int adj:a[u]) consider(adj,adj);
    consider(nxt[u],u);
    ma=max(ma,ma0+ma1);
    return res;
}
ll Work()
{
    int i,u,v;
    ll res=0,ma=-oo;
    for(i=1;i<=ncycle;i++)
    {
        u=c[i];
        c[i]=cal(u,0,res);
    }
    deque <int> dq;
    int j=1;
    for(i=1;i<ncycle;i++)
    {
        while(j<ncycle && sum-dis[j+1]+dis[i]>=dis[j+1]-dis[i])
        {
            j++;
            while(dq.size()>0 && c[dq.back()]-dis[dq.back()]<c[j]-dis[j])
                dq.pop_back();
            dq.push_back(j);
        }
        if(dq.size()>0 && dq.front()==i) dq.pop_front();
        if(dq.size()>0) res=max(res,c[i]+c[dq.front()]+sum+dis[i]-dis[dq.front()]);
    }

    j=ncycle+1;
    for(i=ncycle-1;i>=1;i--)
    {
        while(j-1>i && sum-dis[j-1]+dis[i]<dis[j-1]-dis[i])
        {
            j--;
            ma=max(ma,c[j]+dis[j]);
        }
        res=max(res,c[i]-dis[i]+ma);
    }
    return res;
}
int main()
{
 //   freopen("islands.inp","r",stdin);
  //  freopen("islands.out","w",stdout);
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    cin>>n;
    for(u=1;u<=n;u++)
    {
        cin>>v>>k;
        nxt[u]=v; a[v].push_back(u);
        w[u]=k;
    }
    for(u=1;u<=n;u++)
        if(visited[u]==0)
        {
            dfs(u);
            tr[u]=-1;
            ncycle=0;
            c.clear(); c.push_back(0);
            dis.clear(); dis.push_back(0);
            Find_cycle(u,0);
            res+=Work();
        }
    cout<<res;
}

Compilation message

islands.cpp: In function 'long long int Work()':
islands.cpp:93:13: warning: unused variable 'v' [-Wunused-variable]
   93 |     int i,u,v;
      |             ^
# Verdict Execution time Memory Grader output
1 Correct 7 ms 33116 KB Output is correct
2 Correct 7 ms 33116 KB Output is correct
3 Correct 7 ms 33112 KB Output is correct
4 Correct 7 ms 33116 KB Output is correct
5 Correct 8 ms 33268 KB Output is correct
6 Correct 7 ms 33112 KB Output is correct
7 Correct 7 ms 33116 KB Output is correct
8 Correct 7 ms 33116 KB Output is correct
9 Correct 7 ms 33112 KB Output is correct
10 Correct 7 ms 33368 KB Output is correct
11 Correct 7 ms 33116 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 7 ms 33112 KB Output is correct
2 Correct 7 ms 33124 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 7 ms 33116 KB Output is correct
2 Correct 8 ms 33372 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 12 ms 33884 KB Output is correct
2 Correct 17 ms 35420 KB Output is correct
3 Correct 13 ms 33888 KB Output is correct
4 Correct 10 ms 33548 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 20 ms 36564 KB Output is correct
2 Correct 28 ms 40148 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 54 ms 46468 KB Output is correct
2 Correct 57 ms 55156 KB Output is correct
3 Correct 79 ms 61652 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 88 ms 53584 KB Output is correct
2 Correct 126 ms 70716 KB Output is correct
3 Correct 143 ms 87472 KB Output is correct
4 Correct 171 ms 96120 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 161 ms 53952 KB Output is correct
2 Correct 474 ms 112956 KB Output is correct
3 Correct 167 ms 63312 KB Output is correct
4 Correct 240 ms 96580 KB Output is correct
5 Correct 228 ms 95932 KB Output is correct
6 Correct 828 ms 68656 KB Output is correct
7 Correct 266 ms 131072 KB Output is correct
# Verdict Execution time Memory Grader output
1 Runtime error 220 ms 131072 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -