oj.uz

Submission #489902

# Submission time Handle Problem Language Result Execution time Memory
489902 2021-11-25T03:08:55 Z fhvirus Sirni (COCI17_sirni) C++17
112 / 140
 5000 ms 394828 KB 
// Knapsack DP is harder than FFT.
#include<bits/stdc++.h>
using namespace std;
typedef long long ll; typedef pair<int,int> pii; typedef pair<ll,ll> pll;
#define ff first
#define ss second
#define pb emplace_back
#define AI(x) begin(x),end(x)
template<class I>bool chmax(I&a,I b){return a<b?(a=b,true):false;}
template<class I>bool chmin(I&a,I b){return b<a?(a=b,true):false;} 
#ifdef OWO
#define debug(args...) SDF(#args, args)
#define OIU(args...) ostream& operator<<(ostream&O,args)
#define LKJ(S,B,E,F) template<class...T>OIU(S<T...>s){O<<B;int c=0;for(auto i:s)O<<(c++?", ":"")<<F;return O<<E;}
LKJ(vector,'[',']',i)LKJ(deque,'[',']',i)LKJ(set,'{','}',i)LKJ(multiset,'{','}',i)LKJ(unordered_set,'{','}',i)LKJ(map,'{','}',i.ff<<':'<<i.ss)LKJ(unordered_map,'{','}',i.ff<<':'<<i.ss)
template<class...T>void SDF(const char* s,T...a){int c=sizeof...(T);if(!c){cerr<<"\033[1;32mvoid\033[0m\n";return;}(cerr<<"\033[1;32m("<<s<<") = (",...,(cerr<<a<<(--c?", ":")\033[0m\n")));}
template<class T,size_t N>OIU(array<T,N>a){return O<<vector<T>(AI(a));}template<class...T>OIU(pair<T...>p){return O<<'('<<p.ff<<','<<p.ss<<')';}template<class...T>OIU(tuple<T...>t){return O<<'(',apply([&O](T...s){int c=0;(...,(O<<(c++?", ":"")<<s));},t),O<<')';}
#else
#pragma GCC optimize("Ofast")
#define debug(...) ((void)0)
#endif
 
struct EEK {
	int w, u, v;
	EEK (int w, int u, int v) : w(w), u(u), v(v) {}
	const bool operator < (const EEK &oth) const {
		return w < oth.w;
	}
};
struct DSU {
	int n; vector<int> f;
	DSU (int nn) : n(nn), f(nn) { iota(AI(f), 0); }
	int F(int u) { return u == f[u] ? u : f[u] = F(f[u]); }
};
 
signed main(){
	ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
 
	int N; cin >> N;
	vector<int> P(N); for(int &i: P) cin >> i;
 
	sort(AI(P));
	P.erase(unique(AI(P)), end(P));
	N = P.size();
 
	vector<EEK> e;
	if(N <= 1000){
		for(int i = 0; i < N; ++i)
			for(int j = i+1; j < N; ++j)
				e.pb(min(P[j] % P[i], P[i]), i, j);
	} else {
		assert(P.back() <= 1e7);
		for(int i = 0; i < N-1; ++i){
			for(int j = P[i]; j <= P.back(); j += P[i]){
				int p = lower_bound(begin(P)+i+1, end(P), j) - begin(P);
				int r = P[p] / P[i];
				e.pb(P[p] - P[i] * r, i, p);
				j = P[i] * r;
			}
		}
	}
	sort(AI(e));
 
	DSU dsu(N);
	int ans = 0;
	int cnt = N;
	for(auto [w, u, v]: e){
		u = dsu.F(u);
		v = dsu.F(v);
		if(u != v){
			dsu.f[v] = u;
			ans += w;
			--cnt;
		}
		if(cnt == 1) break;
	}
	cout << ans << '\n';
 
	return 0;
}

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 47 ms 6588 KB Output is correct
2 Correct 36 ms 6588 KB Output is correct
3 Correct 53 ms 6596 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 36 ms 6616 KB Output is correct
2 Correct 18 ms 3520 KB Output is correct
3 Correct 58 ms 6588 KB Output is correct

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 56 ms 6588 KB Output is correct
2 Correct 48 ms 6588 KB Output is correct
3 Correct 54 ms 6588 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 173 ms 13092 KB Output is correct
2 Correct 608 ms 49968 KB Output is correct
3 Correct 253 ms 25404 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 28 ms 3520 KB Output is correct
2 Correct 319 ms 49988 KB Output is correct
3 Correct 166 ms 13208 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 401 ms 50028 KB Output is correct
2 Correct 782 ms 99352 KB Output is correct
3 Correct 210 ms 25388 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 58 ms 6664 KB Output is correct
2 Correct 727 ms 99272 KB Output is correct
3 Correct 210 ms 25368 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 225 ms 25416 KB Output is correct
2 Correct 3583 ms 394824 KB Output is correct
3 Correct 248 ms 25372 KB Output is correct

Subtask #9
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 204 ms 25368 KB Output is correct
2 Execution timed out 5125 ms 394776 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 35 ms 3584 KB Output is correct
2 Execution timed out 5123 ms 394828 KB Time limit exceeded
3 Halted 0 ms 0 KB -