제출 #1263484

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1263484		asdf4321	Bitaro’s Party (JOI18_bitaro)	C++20	100 / 100	914 ms	245572 KiB

bitaro

#include<bits/stdc++.h>
#define endl '\n'
#define fi first
#define se second
using namespace std;

const int MAXN=1e5+5;
const int B=300;

int n,m,q;
pair<int,int> maxd[MAXN][B+5];
vector<int> adj[MAXN];
int dp[MAXN];
int c[MAXN];
bool used[MAXN];

int calc_dp(int v)
{
    for(int i=0; i<=n; i++) dp[i]=-1;
    dp[v]=0;
    for(int i=v-1; i>=1; i--)
    {
        for(int j=0; j<adj[i].size(); j++)
        {
            if(dp[adj[i][j]]==-1) continue;
            dp[i]=max(dp[i], dp[adj[i][j]]+1);
        }
    }
    int ans=-1;
    for(int i=v; i>=1; i--)
    {
        if(!used[i]) ans=max(ans, dp[i]);
    }
    return ans;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin>>n>>m>>q;
    for(int i=0; i<m; i++)
    {
        int u,v;
        cin>>u>>v;
        adj[u].push_back(v);
    }

    for(int i=0; i<=n; i++) maxd[i][0].fi=0, maxd[i][0].se=i;
    for(int i=0; i<=n; i++)
    {
        for(int j=1; j<B; j++) maxd[i][j].fi=-1;
    }

    for(int v=1; v<=n; v++)
    {
        for(int j=0; j<adj[v].size(); j++)
        {
            int to=adj[v][j];
            int ptr1=0;
            int ptr2=0;
            vector<pair<int,int> > tmp;
            for(;;)
            {
                if(tmp.size()>=B) break;
                pair<int,int> cur= {0,0};
                if(maxd[v][ptr1].fi!=-1 && maxd[v][ptr1].fi+1>maxd[to][ptr2].fi)
                {
                    cur= {maxd[v][ptr1].fi+1, maxd[v][ptr1++].se};
                }
                else if(maxd[to][ptr2].fi!=-1)
                {
                    cur= {maxd[to][ptr2].fi, maxd[to][ptr2++].se};
                }
                else break;

                if(!used[cur.se])
                {
                    tmp.push_back(cur);
                    used[cur.se]=1;
                }
            }
            for(int i=0; i<tmp.size(); i++) maxd[to][i]=tmp[i];
            for(int i=0; i<tmp.size(); i++) used[tmp[i].se]=0;
        }
    }

    for(int i=0; i<q; i++)
    {
        int v,cnt;
        cin>>v>>cnt;
        for(int j=0; j<cnt; j++)
        {
            cin>>c[j];
            used[c[j]]=1;
        }

        if(cnt<B)
        {
            bool fl=0;
            for(int j=0; j<B; j++)
            {
                if(maxd[v][j].fi<0) break;
                if(!used[maxd[v][j].se])
                {
                    cout<<maxd[v][j].fi<<endl;
                    fl=1;
                    break;
                }
            }
            if(!fl) cout<<-1<<endl;
        }

        else
        {
            int ans=calc_dp(v);
            cout<<ans<<endl;
        }

        for(int j=0; j<cnt; j++) used[c[j]]=0;
    }

    return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 7 / 7
• Subtask 3: 86 / 86