結果

問題 No.3093 Safe Infection
ユーザー pengin_2000
提出日時 2025-04-06 16:56:53
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 90 ms / 2,000 ms
コード長 2,092 bytes
コンパイル時間 359 ms
コンパイル使用メモリ 28,588 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-04-06 16:56:59
合計ジャッジ時間 5,575 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 70
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function ‘main’:
main.c:90:9: warning: ignoring return value of ‘scanf’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   90 |         scanf("%d %d %d", &n, &m, &k);
      |         ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.c:93:17: warning: ignoring return value of ‘scanf’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   93 |                 scanf("%d", &A[i]);
      |                 ^~~~~~~~~~~~~~~~~~
main.c:96:17: warning: ignoring return value of ‘scanf’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   96 |                 scanf("%d %d", &u[i], &v[i]);
      |                 ^~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff #

#include<stdio.h>
int A[100005];
int u[100005], v[100005];
int h[100005], l, z;
int comp_h(int a, int b)
{
	if (z == 0)
	{
		if (u[h[a]] > u[h[b]])
			return 1;
		else
			return -1;
	}
	else
	{
		if (A[h[a]] > A[h[b]])
			return 1;
		else
			return -1;
	}
}
void swap_h(int a, int b)
{
	int f = h[a];
	h[a] = h[b];
	h[b] = f;
	return;
}
void push(int ne)
{
	h[l] = ne;
	int p = l++;
	for (; p > 0; p = (p - 1) / 2)
		if (comp_h((p - 1) / 2, p) > 0)
			swap_h((p - 1) / 2, p);
	return;
}
int pop()
{
	swap_h(0, --l);
	int p = 0;
	for (;;)
	{
		if (2 * p + 2 < l)
		{
			if (comp_h(2 * p + 1, 2 * p + 2) > 0)
			{
				if (comp_h(p, 2 * p + 2) > 0)
					swap_h(p, 2 * p + 2);
				p = 2 * p + 2;
			}
			else
			{
				if (comp_h(p, 2 * p + 1) > 0)
					swap_h(p, 2 * p + 1);
				p = 2 * p + 1;
			}
		}
		else if (2 * p + 1 < l)
		{
			if (comp_h(p, 2 * p + 1) > 0)
				swap_h(p, 2 * p + 1);
			p = 2 * p + 1;
		}
		else
			break;
	}
	return h[l];
}
int par[100005], max_a[100005];
int root(int n)
{
	if (par[n] != n)
		par[n] = root(par[n]);
	return par[n];
}
void uni(int a, int b)
{
	a = root(a);
	b = root(b);
	par[a] = b;
	if (max_a[b] < max_a[a])
		max_a[b] = max_a[a];
	return;
}
int c[100005];
int main()
{
	int n, m, k;
	scanf("%d %d %d", &n, &m, &k);
	int i;
	for (i = 0; i < n; i++)
		scanf("%d", &A[i]);
	for (i = 0; i < m; i++)
	{
		scanf("%d %d", &u[i], &v[i]);
		u[i]--;
		v[i]--;
		if (A[u[i]] < A[v[i]])
		{
			u[i] ^= v[i];
			v[i] ^= u[i];
			u[i] ^= v[i];
		}
	}
	z = 0;
	l = 0;
	for (i = 0; i < m; i++)
		push(i);
	for (i = 0; i < m; i++)
		c[i] = pop();
	for (i = 0; i < n; i++)
	{
		par[i] = i;
		max_a[i] = A[i];
	}
	c[m] = m;
	u[m] = -1;
	z = 1;
	l = 0;
	for (i = 0; i < n; i++)
		push(i);
	int min, mid, max;
	while (l > 0)
	{
		i = pop();
		min = -1;
		max = m;
		while (max - min > 1)
		{
			mid = (max + min) / 2;
			if (u[c[mid]] < i)
				min = mid;
			else
				max = mid;
		}
		for (; u[c[max]] == i; max++)
		{
			if (max_a[root(v[c[max]])] + k < A[i])
			{
				printf("No\n");
				return 0;
			}
			uni(u[c[max]], v[c[max]]);
		}
	}
	printf("Yes\n");
	return 0;
}
0