제출 #1361783

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1361783		SmuggingSpun	Stranded Far From Home (BOI22_island)	C++20	100 / 100	121 ms	30760 KiB

island

#include<bits/stdc++.h>
#define taskname "B"
using namespace std;
typedef long long ll;
const int lim = 2e5 + 5;
int n, m, a[lim];
vector<int>g[lim];
namespace sub1{
  void solve(){
    for(int i = 1; i <= n; i++){
      priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>>pq;
      vector<bool>vis(n + 1, false);
      vis[i] = true;
      for(int& j : g[i]){
        pq.push(make_pair(a[j], j));
        vis[j] = true;
      }
      ll sum = a[i];
      for(int _ = 1; _ < n; _++){
        if(pq.empty()){
          sum = -1;
          break;
        }
        auto [val, j] = pq.top();
        pq.pop();
        if(val > sum){
          sum = -1;
          break;
        }
        sum += val;
        for(int& k : g[j]){
          if(!vis[k]){
            pq.push(make_pair(a[k], k));
            vis[k] = true;
          }
        }
      }
      cout << (sum == -1 ? 0 : 1);
    }
  }
}
namespace sub2345{
  vector<int>ver[lim];
  bitset<lim>can;
  int lab[lim];
  ll w[lim];
  int find_set(int i){
    while(i != lab[i]){
      i = lab[i] = lab[lab[i]];
    }
    return i;
  }
  void solve(){
    vector<int>p(n);
    iota(p.begin(), p.end(), 1);
    sort(p.begin(), p.end(), [&] (int i, int j){
      return a[i] < a[j];
    });
    memset(lab, 0, sizeof(lab));
    memset(w, 0, sizeof(w));
    for(int& i : p){
      w[i] = a[lab[i] = i];
      ver[i].push_back(i);
      can[i] = true;
      for(int& j : g[i]){
        if(lab[j] != 0 && find_set(i) != find_set(j)){
          int I = find_set(i), J = find_set(j);
          if(w[J] < a[i]){
            for(int& k : ver[J]){
              can[k] = false;
            }
            ver[J].clear();
          }
          if(ver[I].size() < ver[J].size()){
            swap(I, J);
          }
          w[lab[J] = I] += w[J];
          for(int& k : ver[J]){
            ver[I].push_back(k);
          }
        }
      }
    }
    for(int i = 1; i <= n; i++){
      cout << can[i];
    }
  }
}
int main(){
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	if(fopen(taskname".inp", "r")){
		freopen(taskname".inp", "r", stdin);
	}
  cin >> n >> m;
  for(int i = 1; i <= n; i++){
    cin >> a[i];
  }
  for(int i = 0; i < m; i++){
    int u, v;
    cin >> u >> v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  if(max(n, m) <= 2000){
    sub1::solve();
  }
  else{
    sub2345::solve();
  }
}

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

island.cpp: In function 'int main()':
island.cpp:92:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   92 |                 freopen(taskname".inp", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~

• 서브태스크 1: 10 / 10
• 서브태스크 2: 10 / 10
• 서브태스크 3: 15 / 15
• 서브태스크 4: 30 / 30
• 서브태스크 5: 35 / 35