結果

問題 No.17 2つの地点に泊まりたい
ユーザー myanta
提出日時 2017-05-01 13:40:25
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 46 ms / 5,000 ms
コード長 1,395 bytes
コンパイル時間 534 ms
コンパイル使用メモリ 56,832 KB
実行使用メモリ 7,296 KB
最終ジャッジ日時 2024-10-15 01:40:05
合計ジャッジ時間 1,491 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:95:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   95 |                         scanf("%d", &point[i].s);
      |                         ~~~~~^~~~~~~~~~~~~~~~~~~
main.cpp:97:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   97 |                 scanf("%d", &m);
      |                 ~~~~~^~~~~~~~~~
main.cpp:100:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  100 |                         scanf("%d%d%d", &a, &b, &c);
      |                         ~~~~~^~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

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


using namespace std;


typedef long long ll;
typedef pair<int,int> pii;

struct road_t
{
	int to, c;
};


struct point_t
{
	int s;
	vector<road_t> road;
};


multimap<int, pii> que;


int min_u(int& m, int v)
{
	if(m>v || m<0)
	{
		m=v;
		return 1;
	}
	return 0;
}


void push(int c, int p, int s)
{
	que.insert(make_pair(c, make_pair(p, s)));
}


int pop(int& c, int& p, int& s)
{
	if(que.empty()) return 0;
	c=que.begin()->first;
	p=que.begin()->second.first;
	s=que.begin()->second.second;
	que.erase(que.begin());
	return 1;
}


int solve(vector<point_t>& point)
{
	int c=0, p=0, s=0, n=point.size();
	vector<int> dp(n*n, -1);


	push(c, p, s);
	for(;pop(c, p, s);)
	{
		int idx=s*n+p;
		if(min_u(dp[idx], c)==0) continue;

		for(auto r: point[p].road)
		{
			push(c+r.c, r.to, s);
		}
		if(p==0 || p==n-1 || s==n-1 || s==p) continue;

		int ns=p;
		if(s>0) ns=n-1;
		push(c+point[p].s, p, ns);
	}
	return dp.back();
}


int main(void)
{
	int n, m, i, a, b, c;
	vector<point_t> point;

	while(scanf("%d", &n)==1)
	{
		point.clear();
		point.resize(n);

		for(i=0;i<n;i++)
		{
			scanf("%d", &point[i].s);
		}
		scanf("%d", &m);
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d", &a, &b, &c);
			point[a].road.push_back((road_t){b, c});
			point[b].road.push_back((road_t){a, c});
		}

		printf("%d\n", solve(point));
	}

	return 0;
}
0