#609465 (C++14) No.1365 [Cherry 1st Tune] Whose Fault?

• 問題: No.1365 [Cherry 1st Tune] Whose Fault?
• ユーザー: moririn2528_c
• 提出日時: 2021-01-23 05:03:55
• 言語: C++14 (gcc 15.2.0 + boost 1.89.0)
• 結果: AC
• 実行時間: 364 ms / 2,000 ms
• コード長: 3,027 bytes
• コンパイル時間: 848 ms
• コンパイル使用メモリ: 120,728 KB
• 実行使用メモリ: 8,960 KB
• 最終ジャッジ日時: 2026-06-03 10:55:23
• 合計ジャッジ時間: 20,249 ms
• ジャッジサーバーID （参考情報）: judge1_0 / judge2_0

ファイルパターン	結果
other	AC * 46

ソースコード

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<vector>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#include<deque>
#include<iomanip>
#include<tuple>
#include<cassert>
#include<set>
#include<complex>
#include<numeric>
#include<functional>
using namespace std;
typedef long long int LL;
typedef pair<int,int> P;
typedef pair<LL,int> LP;
const int INF=1<<30;
const LL MAX=1e9+7;

void array_show(int *array,int array_n,char middle=' '){
    for(int i=0;i<array_n;i++)printf("%d%c",array[i],(i!=array_n-1?middle:'\n'));
}
void array_show(LL *array,int array_n,char middle=' '){
    for(int i=0;i<array_n;i++)printf("%lld%c",array[i],(i!=array_n-1?middle:'\n'));
}
void array_show(vector<int> &vec_s,int vec_n=-1,char middle=' '){
    if(vec_n==-1)vec_n=vec_s.size();
    for(int i=0;i<vec_n;i++)printf("%d%c",vec_s[i],(i!=vec_n-1?middle:'\n'));
}
void array_show(vector<LL> &vec_s,int vec_n=-1,char middle=' '){
    if(vec_n==-1)vec_n=vec_s.size();
    for(int i=0;i<vec_n;i++)printf("%lld%c",vec_s[i],(i!=vec_n-1?middle:'\n'));
}

typedef class node{
    typedef long double D;
public:
    D a,b,p;
    node(D _a,D _b,D _p):a(_a),b(_b),p(_p){}
    node(){}
    void connect(node n1){
        b+=n1.b;
        a+=n1.a;
        if(p==0 || n1.p==0)p=0;
        else p=p*n1.p/(p+n1.p);
    }
    D calc(){
        return p*powl(a-b,2);
    }
}node;

class union_find_tree{
private:
    static constexpr int N=100005;
    int n;
    queue<int> q1;
    vector<int> parent;
    vector<int> num;
    typedef long double D;
public:
    vector<node> v1;
    void init(){
        parent.assign(n,-1);
        num.assign(n,1);
        v1.assign(n,node());
    }
    union_find_tree(int n_init){
        assert(n_init>=0);
        n=n_init;
        init();
    }
    union_find_tree(){
        n=N;
        init();
    }

    int check_parent(int x){
        assert(x>=0 && x<n);
        if(parent[x]!=-1){
            q1.push(x);
            return check_parent(parent[x]);
        }
        int a;
        while(!q1.empty()){
            a=q1.front(),q1.pop();
            parent[a]=x;
        }
        return x;
    }

    D connect(int x,int y){
        assert(x>=0 && x<n);
        assert(y>=0 && y<n);
        x=check_parent(x),y=check_parent(y);
        if(x==y)return 0;
        if(num[x]>num[y])swap(x,y);
        parent[x]=y;
        num[y]+=num[x];
        D s=-v1[x].calc()-v1[y].calc();
        v1[y].connect(v1[x]);
        return s+v1[y].calc();
    }

    int size(int pos){
        pos=check_parent(pos);
        return num[pos];
    }
};


int main(){
    int n,m;
    int i,j,k;
    int a,b,c;
    cin>>n;
    union_find_tree ua(n);
    for(i=0;i<n;i++)cin>>ua.v1[i].a;
    for(i=0;i<n;i++)cin>>ua.v1[i].b;
    for(i=0;i<n;i++)cin>>ua.v1[i].p;
    long double s=0;
    for(i=0;i<n;i++)s+=ua.v1[i].calc();
    cin>>m;
    cout<<fixed<<setprecision(20);
    for(i=0;i<m;i++){
        cin>>a>>b;
        a--,b--;
        s+=ua.connect(a,b);
        cout<<s<<endl;
    }
}