제출 #1350586

#제출 시각아이디문제언어결과실행 시간메모리
1350586simplemind_31LOSTIKS (INOI20_lostiks)C++20
23 / 100
311 ms946608 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=1000000;
bool xd=true;
int visited[MAXN+1];
vector<pair<ll,ll>> graph[MAXN+1];
ll s,f,n,a,b,c,m,tiempo,dist[25][MAXN+1],res[(1<<(22))*22+22+1],val[MAXN+1],pad[MAXN+1],necesito[MAXN+1],correspond[MAXN+1];
vector<int> block;
priority_queue<pair<ll,ll>,vector<pair<ll,ll>>,greater<pair<ll,ll>>> dij;
void dfs1(ll node){
    for(auto u:graph[node]){
        if(u.first==pad[node])continue;
        val[u.first]=u.second;
        pad[u.first]=node;
        if(u.second!=-1){
            block.push_back(u.first);
            correspond[u.first]=block.size()-1;
        }
        dfs1(u.first);
    }
}
void dfs2(ll node){
    // para desbloquear cada uno necesito prerequisitos(necesito)
    // necesito desbloquear al padre y al actual
    if(visited[node]==1)xd=false;
    if(visited[node]!=0)return;
    visited[node]=1;
    if(pad[node]!=-1){
        dfs2(pad[node]);
        if(!xd)return;
        necesito[node]|=necesito[pad[node]];
        if(val[pad[node]]!=-1)necesito[node]|=1<<correspond[pad[node]];
    }
    if(!xd)return;
    if(val[node]!=-1){
        // necesito desbloquear val[node]
        dfs2(val[node]);
        if(!xd)return;
        necesito[node]|=necesito[val[node]];
        if(val[val[node]]!=-1)necesito[node]|=1<<correspond[val[node]];
        // si necesito bloq antes para desbloquear bloq->absurd;pero si no esta en el camino?
        if(necesito[node]&(1<<correspond[node])){
            xd=false;
            return;
        }
    }
    visited[node]=2;
}
void dfs3(ll node,ll ind,ll ante){
    for(auto u:graph[node]){
        if(u.first==ante)continue;
        dist[ind][u.first]=dist[ind][node]+1;
        dfs3(u.first,ind,node);
    }
}
// minima cantidad para desboquear subset y terminar en block[i]?
// a que block puedo desbloquear si tengo actual?2^m*m
// tengo todos los requisitos para desbloquear el que tengo que desbloquear 
//dijkstra?
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin >> n >> s >> f;
    s--;f--;
    for(int i=0;i<=n;i++)correspond[i]=val[i]=pad[i]=-1;
    for(int i=1;i<n;i++){
        cin >> a >> b >> c;
        a--;b--;c--;
        if(a==f && c==-1)c=b;
        if(b==f && c==-1)c=a;
        graph[a].push_back({b,c});
        graph[b].push_back({a,c});
    }
    // añado uno extra
    graph[n].push_back({s,n});
    graph[s].push_back({n,n});
    dfs1(n);
    m=block.size();
    int now=f;
    dfs2(f);
    if(s==f){
        cout << 0;
        return 0;
    }
    if(!xd){
        cout << -1;
        return 0;
    }
    //cout << '\n';
    //calcular distancia desde cada block[j] a todos;
    for(ll i=0;i<m;i++){
        dfs3(pad[block[i]],i,-1);
        //for(ll j=0;j<=n;j++)cout << dist[i][j] << ' ';
        //cout << endl;
    }
    for(int i=0;i<=(1<<m)*m+m;i++)res[i]=1e18;
    // desbloquea subset x y termina en el nodo y
    // nuestra respuesta es 
    // caso base, para desbloquear solo el start me cuesta 0
    // termino en padre del block[i];
    dij.push({0,(1<<0)*m+0});
    res[(1<<0)*m+0]=0;
    //cout << endl << endl;
    while(!dij.empty()){
        ll top=dij.top().second;
        ll nada=dij.top().first;
        dij.pop();
        if(res[top]<nada)continue;
        ll x=top/m,y=top%m;
        /*for(ll i=0;i<m;i++){
            if(x&(1<<i))cout << i << ' ';
        }
        cout << ": " << y << ' ';
        cout << res[top] << '\n';*/
        if(y==correspond[f]){
            cout << res[top];
            return 0;
        }
        for(ll j=0;j<m;j++){
            if(x&(1<<j))continue;
            //necesito necesito[block[j]] y tengo x
            if(necesito[block[j]]-(necesito[block[j]]&x))continue;
            // no era posible porwue necesitaba mas cosas
            // ir desde block[y] al val[block[j]] y de val[block[j]] al block[j];
            ll temp=dist[y][val[block[j]]]+dist[j][val[block[j]]]+res[top];
            if(temp<res[(x|(1<<j))*m+j]){
                res[(x|(1<<j))*m+j]=temp;
                //for(ll k=0;)
                /*for(ll i=0;i<m;i++){
                    if((x)&(1<<i))cout << block[i] << ' ';
                }*/
                //cout << ": " << block[y] << ' ' << block[j] << ' ';
                //cout << temp << '\n';
                dij.push({res[(x|(1<<j))*m+j],(x|(1<<j))*m+j});
            }
        }
    }
    cout << -1;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...