oj.uz

답안 #84232

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
84232 2018-11-14T02:33:38 Z ToMoClone Cities (BOI16_cities) C++17
74 / 100
 4069 ms 263168 KB 
/*input
4 3 6
1 3 4
1 2 4
1 3 9
1 4 6
2 3 2
2 4 5
3 4 8
*/
#include <bits/stdc++.h>
using namespace std;

const int N = 100005;
const int64_t inf = 1e15;

struct State {
	int node; int64_t d; int mask;
	State(int _node = 0, int64_t _d = inf, int _mask = 0) :
		node(_node), d(_d), mask(_mask) {}
	bool operator<(const State &x) const {
		return d > x.d;
	}
};

int n, k, m, mark[N];
int64_t d[N][32];
priority_queue<State> q;
vector<pair<int, int> > G[N];

int32_t main(){
	memset(mark, -1, sizeof mark), memset(d, 127, sizeof d);

	scanf("%d%d%d", &n, &k, &m);
	for(int i = 0; i < k; ++ i) {
		int a; scanf("%d", &a); mark[a] = i;
		d[a][1 << i] = 0, q.push(State(a, 0, 1 << i));
	}
	for(int i = 1; i <= m; ++ i) {
		int u, v, c; scanf("%d%d%d", &u, &v, &c);
		G[u].push_back(make_pair(c, v));
		G[v].push_back(make_pair(c, u));
	}

	while(!q.empty()) {
		auto cur = q.top(); q.pop();
		if(cur.mask == (1 << k) - 1) return cout << cur.d << endl, 0;
		for(auto u : G[cur.node])
			if(d[u.second][cur.mask] > d[cur.node][cur.mask] + u.first) {
				d[u.second][cur.mask] = d[cur.node][cur.mask] + u.first;
				q.push(State(u.second, d[u.second][cur.mask], cur.mask));
			}
		int otherMask = ((1 << k) - 1) ^ cur.mask; 
		for(auto u : G[cur.node]) for(int mask = otherMask; mask; mask = otherMask & (mask - 1)) {
			if(d[cur.node][mask | cur.mask] > cur.d + d[u.second][mask] + u.first) {
				d[cur.node][mask | cur.mask] = cur.d + d[u.second][mask] + u.first;
				q.push(State(cur.node, d[cur.node][mask | cur.mask], mask | cur.mask));
			}
			if(d[u.second][mask | cur.mask] > cur.d + d[u.second][mask] + u.first) {
				d[u.second][mask | cur.mask] = cur.d + d[u.second][mask] + u.first;
				q.push(State(u.second, d[u.second][mask | cur.mask], mask | cur.mask));
			}
		}
	}
}

Compilation message

cities.cpp: In function 'int32_t main()':
cities.cpp:34:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d%d%d", &n, &k, &m);
  ~~~~~^~~~~~~~~~~~~~~~~~~~~~
cities.cpp:36:15: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   int a; scanf("%d", &a); mark[a] = i;
          ~~~~~^~~~~~~~~~
cities.cpp:40:21: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   int u, v, c; scanf("%d%d%d", &u, &v, &c);
                ~~~~~^~~~~~~~~~~~~~~~~~~~~~

Subtask #1
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 25 ms 28152 KB Output is correct
2 Correct 26 ms 28216 KB Output is correct
3 Correct 26 ms 28240 KB Output is correct
4 Correct 26 ms 28444 KB Output is correct
5 Correct 26 ms 28444 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 805 ms 63868 KB Output is correct
2 Correct 949 ms 67572 KB Output is correct
3 Correct 197 ms 67572 KB Output is correct
4 Correct 168 ms 67572 KB Output is correct
5 Correct 357 ms 67572 KB Output is correct
6 Correct 116 ms 67572 KB Output is correct
7 Correct 31 ms 67572 KB Output is correct
8 Correct 28 ms 67572 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 36 ms 67572 KB Output is correct
2 Correct 36 ms 67572 KB Output is correct
3 Correct 29 ms 67572 KB Output is correct
4 Correct 34 ms 67572 KB Output is correct

Subtask #4
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4069 ms 159836 KB Output is correct
2 Correct 3477 ms 163516 KB Output is correct
3 Correct 1439 ms 163516 KB Output is correct
4 Correct 2592 ms 169772 KB Output is correct
5 Correct 613 ms 169772 KB Output is correct
6 Correct 325 ms 169772 KB Output is correct

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2487 ms 263168 KB Execution killed with signal 9 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -