글 읽기 - 게시판에 있는 반례에서는 제 코드의 반례를 못 찾고 있습니다

rosie0707 1년 전 0

1967번에서 통과했던 코드 조금 수정해서 제출했는데 틀렸다고 뜹니다....

#include <iostream>
#include <vector>
using namespace std;

int result;

struct node {
	int number;
	int dist;
	int bottomD;
};

vector<node> A_L[100001];
int n;

void get_input()
{
	scanf("%d", &n);

	int lineNum = 0;
	int a, w;
	for (int i = 0; i < n; i++)
	{
		scanf("%d", &lineNum);
		scanf("%d", &a);
		while (a != -1)
		{
			scanf("%d", &w);
			A_L[lineNum].push_back({ a, w, 0 });
			scanf("%d", &a);
		}
	}
}

void get_bottom(int idx, node* parent, int pidx)
{
	bool leaf = true;
	node* curr;
	for (int i = 0; i < A_L[idx].size(); i++)
	{
		curr = &A_L[idx][i];
		if (curr->number == pidx)
			continue;
		leaf = false;
		get_bottom(curr->number, curr, idx);
		if (parent->bottomD < parent->dist + curr->bottomD)
			parent->bottomD = parent->dist + curr->bottomD;
	}
	if (leaf)
		parent->bottomD = parent->dist;
}

int main()
{
	get_input();
	A_L[0].push_back({ 0, 0, 0 });
	get_bottom(1, &A_L[0][0], 0);

	result = 0;
	for (int i = 1; i <= n; i++)
	{
		if (A_L[i].size() == 0)
			continue;

		int tmp = A_L[i][0].bottomD;
		if (result < tmp)
			result = tmp;

		//size more or same than 2
		for (int j = 0; j < A_L[i].size() - 1; j++)
		{
			for (int k = 1; k < A_L[i].size(); k++)
			{
				tmp = A_L[i][j].bottomD + A_L[i][k].bottomD;
				if (result < tmp)
					result = tmp;
			}
		}
	}

	printf("%d", result);

	return 0;
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

게시판에 있는 반례에서는 제 코드의 반례를 못 찾고 있습니다

rosie0707 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말