Submission #345144

#	Time	Username	Problem	Language	Result	Execution time	Memory
345144		casperwang	Tropical Garden (IOI11_garden)	C++14	69 / 100	435 ms	55136 KiB

garden

#include "garden.h"
#include "gardenlib.h"
#include <bits/stdc++.h>
#define All(x) x.begin(), x.end()
#define pb emplace_back
#define pii pair<int,int>
#define ff first
#define ss second
using namespace std;
#define debug(args...) kout("[ " + string(#args) + " ]", args)
void kout() { cerr << endl; }
template <class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ',kout(b...); }
template <class T> void pary(T L, T R) { while (L != R) cerr << *L << " \n"[++L==R]; }

const int MAXN = 150000;
const int MAXQ = 2000;
const int L = 30;

void count_routes(int N, int M, int P, int R[MAXN][2], int Q, int G[MAXQ]) {
  vector <vector<int>> path(N);
  for (int i = 0; i < M; i++) {
    path[R[i][0]].pb(R[i][1]);
    path[R[i][1]].pb(R[i][0]);
  }
  vector <vector<int>> anc(2*N);
  for (int i = 0; i < N; i++) {
    anc[i].resize(L);
    anc[i+N].resize(L);
    int t = path[i][0];
    anc[i][0] = (path[t][0] == i && path[t].size() > 1) ? t+N : t;
    if (path[i].size() > 1) {
      t = path[i][1];
      anc[i+N][0] = (path[t][0] == i && path[t].size() > 1) ? t+N : t;
    } else anc[i+N][0] = i+N;
  }
  for (int i = 1; i < L; i++)
    for (int j = 0; j < 2*N; j++)
      anc[j][i] = anc[anc[j][i-1]][i-1];
  vector <int> cur(N);
  for (int i = 0; i < N; i++)
    cur[i] = i;
  for (int q = 0; q < Q; q++) {
    int ans = 0;
    for (int i = 0; i < N; i++) {
      int dis = G[q] - (q ? G[q-1] : 0);
      for (int j = 0; j < L; j++) {
        if (dis & (1<<j)) cur[i] = anc[cur[i]][j];
      }
      if (cur[i] % N == P) ans++;
    }
    answer(ans);
  }
  return;
}

• Subtask 1: 49 / 49
• Subtask 2: 20 / 20
• Subtask 3: 0 / 31