QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#626340#4231. Rafting Tripnickbelov#RE 0ms0kbC++203.1kb2024-10-10 04:58:342024-10-10 04:58:34

Judging History

你现在查看的是最新测评结果

  • [2024-10-10 04:58:34]
  • 评测
  • 测评结果:RE
  • 用时:0ms
  • 内存:0kb
  • [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##^
#<<<.^

output:


result: