QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#626340 | #4231. Rafting Trip | nickbelov# | RE | 0ms | 0kb | C++20 | 3.1kb | 2024-10-10 04:58:34 | 2024-10-10 04:58:34 |
answer
#include <vector>
#pragma GCC target ("avx2")
#pragma GCC optimize ("O3")
#pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
template<typename T>
ostream_iterator<T> oit(const string &s = " "){ return ostream_iterator<T>(cout,s.c_str()); }
inline auto rep(int l, int r) { return views::iota(min(l, r), r); }
inline auto rep(int n) { return rep(0, n); }
inline auto rep1(int l, int r) { return rep(l, r + 1); }
inline auto rep1(int n) { return rep(1, n + 1); }
inline auto per(int l, int r) { return rep(l, r) | views::reverse; }
inline auto per(int n) { return per(0, n); }
inline auto per1(int l, int r) { return per(l, r + 1); }
inline auto per1(int n) { return per(1, n + 1); }
#define A(a) begin(a),end(a)
inline auto len = ranges::ssize;
struct chash {
static uint64_t splitmix64(uint64_t x) {
// http://xorshift.di.unimi.it/splitmix64.c
x += 0x9e3779b97f4a7c15;
x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
return x ^ (x >> 31);
}
size_t operator()(uint64_t x) const {
static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
return splitmix64(x + FIXED_RANDOM);
}
};
template<typename T, typename U> using pb_map = gp_hash_table<T, U, chash>;
template<typename T> using pb_set = gp_hash_table<T, null_type, chash>;
#define K first
#define V second
using ll = long long;
using ld = long double;
using vi = vector<int>;
using vii = vector<vector<int>>;
typedef vector<ll> vll;
using pll = pair<ll,ll>;
using pii = pair<int,int>;
constexpr ll NN = 5000,M = 200;
vi nm(101);
int dp[NN][M];
vi adj[NN];
int a[NN];
void dfs(int i,int p){
if(p != -1) adj[i].erase(ranges::find(adj[i],p));
for(int j : adj[i])
dfs(j,i);
for(int here=2;here<M;here++){
if(a[i]!=here) dp[i][here]=here;
for(int j : adj[i]){
auto &v = dp[j];
int add = v[here];
for(int c=2;c<M;c++){
if((((nm[c]&1)==0) and ((nm[here]&1)==0)) or (gcd(nm[c],nm[here])>1))
add=min(add,v[c]);
}
dp[i][here]+=add;
}
}
}
void run()
{
int n; cin >> n;
for(int i : rep(n)) cin >> a[i];
for(int u,v; int i : rep(n-1)){
cin >> u >> v,--u,--v;
adj[u].push_back(v),adj[v].push_back(u);
} dfs(0,-1);
int ans = dp[0][2];
for(int c : rep(2,M)) ans = min(ans,dp[0][c]);
cout << ans << '\n';
}
int main()
{
//KING OF THE WORLD...... U.W.T.B
cin.tie(0);
ios_base::sync_with_stdio(false);
for(int i : rep1(100)) nm[i]=i;
for(int x : rep1(101,250)){
int nx = x;
vi f;
for(int i = 2;i*i<=nx;i++){
if(nx%i==0)f.push_back(i);
while(nx%i==0) nx/=i;
} if(nx>1) f.push_back(nx);
if(ranges::max(f)<=100) nm.push_back(x);
}
run();
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Runtime Error
input:
5 6 v<<<#v v#v<.> >>v<<< ..v##^ #<<<.^