제출 #592296

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
592296		APROHACK	악어의 지하 도시 (IOI11_crocodile)	C++14	46 / 100	5 ms	3540 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define ff first
#define ss second
ll const NMX = 100001;
bool vis[NMX];
vector<pair<ll, int> >adj[NMX];
int bfsans[NMX];
int n;

void printBFSANS(){
  cout << "bfs number\n";
  for(int i = 0 ; i < n ; i++){

    cout << bfsans[i] << " ";
  }
  cout<<endl;
}
pair<ll, ll>dp[NMX];
pair<ll, ll>answer(int node){
  //cout<<" llegue "<<node<<endl;
  if(bfsans[node]==0)return dp[node]={0, 0};
  vis[node]=true;
  ll minimum1=LLONG_MAX, minimum2=LLONG_MAX;
  for(int i = 0 ; i < adj[node].size() ; i++){
    if(bfsans[adj[node][i].ss] == -1 || vis[adj[node][i].ss] || bfsans[adj[node][i].ss] > bfsans[node])continue;
    ll v;
    if(dp[adj[node][i].ss].ff == -1){
      pair<ll, ll>cur = answer(adj[node][i].ss);
      v = max(cur.ff, cur.ss) + adj[node][i].ff;
      if(cur.ff == -1 || cur.ss == -1)continue;
      
    }else{
      v = max(dp[adj[node][i].ss].ff, dp[adj[node][i].ss].ss) +  adj[node][i].ff;
    }
    
    if(v<minimum1){
        minimum2=minimum1;
        minimum1=v;
      }else if(v<minimum2){
        minimum2=v;
      }
  }
  vis[node]=false;
  if(minimum1==LLONG_MAX || minimum2 == LLONG_MAX){
    bfsans[node]=-1;
    return dp[node] = {-1, -1};
  }
  dp[node].ff=minimum1, dp[node].ss=minimum2;
  //cout<<"node: "<<node << " "<<minimum1 << " " << minimum2 << endl;
  return dp[node];
}

int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
  memset(vis, false, sizeof vis);
  n=N;
  for(int i = 0 ; i <M  ; i++){
    adj[R[i][0]].pb({L[i], R[i][1]});
    adj[R[i][1]].pb({L[i], R[i][0]});
  }
  for(int i = 0 ; i < n ; i++)bfsans[i]=-1;
  queue<pair<ll, ll> >nodos;
  for(int i = 0 ; i < K ; i ++){
    nodos.push({0, P[i]});
    bfsans[P[i]]=0;
  }
  while(!nodos.empty()){
    pair<ll, ll> cur = nodos.front();
    nodos.pop();
    if(vis[cur.ss] )continue;
    ll cuenta = 0;
   // cout<<cur.ss << " ";
    for(int i = 0 ; i < adj[cur.ss].size() ; i++){
      if(bfsans[ adj[cur.ss][i].ss ] != -1)cuenta++;
      if(!vis[adj[cur.ss][i].ss]){
       // vis[adj[cur.ss][i].ss] = true;
        nodos.push({cur.ff+1, adj[cur.ss][i].ss});
        //cout << cur.ff+1 << " ";
        }
    }
    //cout<<endl;
    if(cuenta>=2){
      bfsans[cur.ss]=cur.ff;
      vis[cur.ss]=true;
    }
    if(bfsans[cur.ss]!=-1)vis[cur.ss]=true;
  }
  memset(vis, false, sizeof vis);
  //printBFSANS();
  for(int i = 0 ; i < n ; i ++){
    dp[i]={-1, -1};
  }
  pair<ll, ll>retu = answer(0);
  return max(retu.ff, retu.ss);
}

컴파일 시 표준 에러 (stderr) 메시지

crocodile.cpp: In function 'std::pair<long long int, long long int> answer(int)':
crocodile.cpp:28:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |   for(int i = 0 ; i < adj[node].size() ; i++){
      |                   ~~^~~~~~~~~~~~~~~~~~
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:77:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   77 |     for(int i = 0 ; i < adj[cur.ss].size() ; i++){
      |                     ~~^~~~~~~~~~~~~~~~~~~~

• Subtask 1: 46 / 46
• Subtask 2: 0 / 43
• Subtask 3: 0 / 11