結果

問題 No.417 チューリップバブル
ユーザー myanta
提出日時 2017-05-31 08:23:54
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,419 bytes
コンパイル時間 587 ms
コンパイル使用メモリ 60,696 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-21 20:16:42
合計ジャッジ時間 1,652 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6 WA * 34
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:93:43: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   93 |                 for(int i=0;i<n;i++) scanf("%d", &u[i]);
      |                                      ~~~~~^~~~~~~~~~~~~
main.cpp:99:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   99 |                         scanf("%d%d%d", &a, &b, &c);
      |                         ~~~~~^~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include<cstdio>
#include<vector>
#include<queue>
#include<map>


using namespace std;
using vi=vector<int>;
using vvi=vector<vi>;
using pii=pair<int,int>;
using vpii=vector<pii>;
using vvpii=vector<vpii>;


struct q_t
{
	int p, r, c;
	vi f;
	bool operator<(const q_t& rhs) const
	{
		return c>rhs.c;
	}
};


class pq_t
{
	priority_queue<q_t> q;
public:
	void push(int p, int r, int c, vi& f)
	{
		q.push((q_t){p, r, c, f});
	}
	int pop(int&p, int&r, int&c, vi&f)
	{
		if(q.empty()) return 0;
		auto t=q.top();
		q.pop();
		p=t.p;
		r=t.r;
		c=t.c;
		f=t.f;
		return 1;
	}
};


int solve(vvpii& v, vi& u, int m)
{
	int n=v.size();
	vi f;
	int p=0, r=u[0], c=m, ret=-1;
	pq_t q;

	f.resize(n);

	f[0]=1;
	q.push(p, r, c, f);
	while(q.pop(p, r, c, f))
	{
//printf("p=%d r=%d c=%d\n", p, r, c);
		if(c<0) break;
		if(ret<r) ret=r;

		for(int i=0;i<v[p].size();i++)
		{
			int np, nr, nc;
			np=v[p][i].first;
			nc=c-v[p][i].second*2;
			if(f[np]) continue;

			f[np]=1;
			nr=r+u[np];
			q.push(np, nr, nc, f);
			f[np]=0;
		}
	}
	return ret;
}


int main(void)
{
	int n, m;
	vi u;
	vvpii v;

	while(scanf("%d%d", &n, &m)==2)
	{
		v.clear();
		v.resize(n);
		u.resize(n);
		for(int i=0;i<n;i++) scanf("%d", &u[i]);

		for(int i=0;i<n-1;i++)
		{
			int a, b, c;

			scanf("%d%d%d", &a, &b, &c);
			v[a].push_back(make_pair(b, c));
			v[b].push_back(make_pair(a, c));
		}
		printf("%d\n", solve(v, u, m));
	}

	return 0;
}
0