QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#691499 | #2685. Driving in Optimistan | kevinyang | WA | 10ms | 7856kb | C++17 | 4.1kb | 2024-10-31 11:41:02 | 2024-10-31 11:41:02 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define double long double
#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
int f(int l, int r, int k){
// if(l > r){
// swap(l,r);
// }
return k*l*r + (r-l)*k*(k-1)/2 - (k-1)*k*(2*k-1)/6;
}
int f2(int l, int r, int k){
int ans = 0;
if(l > r){
swap(l,r);
}
for(int i = 0; i<k; i++){
ans+=(l+i)*(r-i);
}
return ans;
}
signed main() {
cin.tie(0)->sync_with_stdio(0);
int n;
cin >> n;
vector<vector<int>>d(n+1,vector<int>(n+1));
for(int i = 1; i<=n; i++){
for(int j = i+1; j<=n; j++){
cin >> d[i][j];
d[j][i] = d[i][j];
}
}
vector<vector<int>>dep(n+1,vector<int>(n+1)); // dep[i][j] is the depth of the lca(i,j)
for(int i = 2; i<=n; i++){
for(int j = 2; j<=n; j++){
if(i==j)continue;
if((d[1][i] + d[1][j] - d[i][j])%2 !=0){
cout << "impossible\n";
return 0;
}
dep[i][j] = (d[1][i] + d[1][j] - d[i][j])/2;
}
}
vector<vector<pii>>adj(2*n+5);
int label = n+1;
int tot = 0;
auto solve = [&](auto self, vector<int>nodes, int parent, int w) -> void {
int mn = 1e9;
int m = nodes.size();
if(nodes.size() == 0)return;
if(nodes.size() == 1){
int u = nodes[0];
int depth = d[1][u];
adj[parent].push_back({depth - w, u});
adj[u].push_back({depth - w,parent});
tot += depth-w;
return;
}
int cur = label++;
vector<vector<int>>buckets(m);
for(int i = 0; i<nodes.size(); i++){
for(int j = i+1; j<nodes.size(); j++){
mn = min(mn,dep[nodes[i]][nodes[j]]);
}
}
tot += mn-w;
adj[parent].push_back({mn-w,cur});
adj[cur].push_back({mn-w,parent});
for(int u : nodes){
//cout << "nodes " << u << '\n';
for(int i = 0; i<m; i++){
if(buckets[i].size() == 0){
buckets[i].push_back(u);
break;
}
if(dep[buckets[i][0]][u] != mn){
buckets[i].push_back(u);
break;
}
}
}
for(auto vec: buckets){
self(self, vec, cur, mn);
}
};
vector<int>vec;
for(int i = 2; i<=n; i++){
vec.push_back(i);
}
solve(solve,vec,1,0);
// for(int i = 1; i<=2*n; i++){
// for(auto [w,y] : adj[i]){
// if(i<y)cout << i << ' ' << y << ' ' << w << '\n';
// }
// }
for(int i = 1; i<=n; i++){
vector<int>dis(2*n+1);
vector<bool>vis(2*n+1);
vis[i] = true;
queue<int>q;
q.push(i);
while(q.size()){
int cur = q.front(); q.pop();
for(auto [w,nxt]: adj[cur]){
if(vis[nxt])continue;
vis[nxt] = true;
dis[nxt] = dis[cur] + w;
q.push(nxt);
}
}
for(int j = 1; j<=n; j++){
if(d[i][j] != dis[j]){
cout << "impossible\n";
return 0;
}
}
}
double ans = 0;
vector<int>dp(2*n+1);
auto dfs = [&](auto self, int u, int p, int w) -> void {
for(auto [wt, nxt] : adj[u]){
if(nxt==p)continue;
self(self,nxt,u,wt);
dp[u]+=dp[nxt];
dp[u]+=wt;
}
//cout << u << ' ' << p << ' '<< dp[u]+1 << ' ' << tot-dp[u] << ' ' << w << '\n';
if(w > 0){
ans += f(dp[u]+1,tot-dp[u],w);
}
};
dfs(dfs,1,0,0);
ans /= (double)tot * (tot+1)/2;
cout << fixed << setprecision(15);
cout << ans << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3816kb
input:
4 4 4 4 2 4 4
output:
2.392857142857143
result:
ok
Test #2:
score: 0
Accepted
time: 10ms
memory: 7856kb
input:
500 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30000 30...
output:
14980.998002622488539
result:
ok
Test #3:
score: 0
Accepted
time: 0ms
memory: 4044kb
input:
4 8 8 8 8 8 8
output:
4.117647058823529
result:
ok
Test #4:
score: -100
Wrong Answer
time: 7ms
memory: 7824kb
input:
500 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 1000000 ...
output:
56612.475337856860204
result:
wrong answer