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

• 問題: No.1365 [Cherry 1st Tune] Whose Fault?
• ユーザー: publfl
• 提出日時: 2021-01-22 21:52:06
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 307 ms / 2,000 ms
• コード長: 1,373 bytes
• コンパイル時間: 321 ms
• コンパイル使用メモリ: 33,664 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-09 06:51:23
• 合計ジャッジ時間: 16,523 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	27 ms; 5,376 KB
testcase_14	AC	19 ms; 5,376 KB
testcase_15	AC	6 ms; 5,376 KB
testcase_16	AC	5 ms; 5,376 KB
testcase_17	AC	16 ms; 5,376 KB
testcase_18	AC	10 ms; 5,376 KB
testcase_19	AC	26 ms; 5,376 KB
testcase_20	AC	16 ms; 5,376 KB
testcase_21	AC	24 ms; 5,376 KB
testcase_22	AC	20 ms; 5,376 KB
testcase_23	AC	61 ms; 5,376 KB
testcase_24	AC	124 ms; 5,376 KB
testcase_25	AC	246 ms; 5,376 KB
testcase_26	AC	168 ms; 5,376 KB
testcase_27	AC	202 ms; 5,376 KB
testcase_28	AC	84 ms; 5,376 KB
testcase_29	AC	71 ms; 5,376 KB
testcase_30	AC	307 ms; 5,376 KB
testcase_31	AC	280 ms; 5,376 KB
testcase_32	AC	154 ms; 5,376 KB
testcase_33	AC	239 ms; 5,376 KB
testcase_34	AC	189 ms; 5,376 KB
testcase_35	AC	114 ms; 5,376 KB
testcase_36	AC	231 ms; 5,376 KB
testcase_37	AC	50 ms; 5,376 KB
testcase_38	AC	67 ms; 5,376 KB
testcase_39	AC	287 ms; 5,376 KB
testcase_40	AC	282 ms; 5,376 KB
testcase_41	AC	216 ms; 5,376 KB
testcase_42	AC	196 ms; 5,376 KB
testcase_43	AC	2 ms; 5,376 KB
testcase_44	AC	83 ms; 5,376 KB
testcase_45	AC	99 ms; 5,376 KB

ソースコード

#include <stdio.h>

int next[100010];
int find(int k)
{
	if(next[k]==k) return k;
	else return next[k] = find(next[k]);
}

int A[100010],B[100010];
double P[100010];
int check[100010];

int main()
{
	int a;
	scanf("%d",&a);
	for(int i=1;i<=a;i++) scanf("%d",&A[i]);
	for(int i=1;i<=a;i++) scanf("%d",&B[i]);
	for(int i=1;i<=a;i++)
	{
		int c;
		scanf("%d",&c);
		if(c==0) check[i] = 1;
		else P[i] = (double)1/(2*c);
	}
	
	double sum = 0;
	for(int i=1;i<=a;i++) next[i] = i;
	for(int i=1;i<=a;i++)
	{
		if(check[i]==0)
		{
			long long int lambda = (A[i]-B[i]);
			sum += (lambda*lambda)/(2*P[i]);
		}
	}
	
	int b;
	scanf("%d",&b);
	for(int i=1;i<=b;i++)
	{
		int c,d;
		scanf("%d%d",&c,&d);
		c = find(c);
		d = find(d);
		if(c!=d)
		{
			long long int lambda;
			if(check[c]==1&&check[d]==1);
			else if(check[c]==1)
			{
				lambda = (A[d]-B[d]);
				sum -= (lambda * lambda)/(2*P[d]);
				next[c] = d;
				check[d] = 1;
			}
			else if(check[d]==1)
			{
				lambda = (A[c]-B[c]);
				sum -= (lambda * lambda)/(2*P[c]);
				next[c] = d;
			}
			else
			{
				lambda = (A[c]-B[c]);
				sum -= (lambda * lambda)/(2*P[c]);
				lambda = (A[d]-B[d]);
				sum -= (lambda * lambda)/(2*P[d]);
				
				next[c] = d;
				A[d] += A[c];
				B[d] += B[c];
				P[d] += P[c];
				
				lambda = (A[d]-B[d]);
				sum += (lambda * lambda)/(2*P[d]);
			}
		}
		printf("%.12lf\n",sum);
	}
}