oj.uz

Submission #56207

# Submission time Handle Problem Language Result Execution time Memory
56207 2018-07-10T08:57:47 Z 윤교준(#1582) Cities (BOI16_cities) C++11
74 / 100
 6000 ms 172636 KB 
#include <bits/stdc++.h>
#define rf(x) (x)=0;while(*p<48)p++;while(47<*p)(x)=((x)<<3)+((x)<<1)+(*p++&15);
#define pb push_back
#define eb emplace_back
#define sz(V) ((int)(V).size())
#define allv(V) ((V).begin()),((V).end())
#define befv(V) ((V)[(sz(V)-2)])
#define sorv(V) sort(allv(V))
#define revv(V) reverse(allv(V))
#define univ(V) (V).erase(unique(allv(V)),(V).end())
#define clv(V) (V).clear()
#define upmin(a,b) (a)=min((a),(b))
#define upmax(a,b) (a)=max((a),(b))
#define rb(x) ((x)&(-(x)))
#define INF (0x3f3f3f3f)
#define INFLL (0x3f3f3f3f3f3f3f3fll)
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
static unsigned char str[22000022], *p=str;
void fg(vector<pii> G[], int a, int b, int c) { G[a].eb(b, c); G[b].eb(a, c); }
const int MAXN = 100005;
const int MAXM = 200005;

priority_queue<pli, vector<pli>, greater<pli>> PQ;

vector<pii> G[MAXN];

ll dp[MAXN][1<<5];

int A[MAXM], B[MAXM], C[MAXM];
int D[MAXN];

ll Ans = INFLL;
int N, M, K;

int main() {
	fread(str, 1, 22000022, stdin);
	rf(N) rf(K) rf(M)

	for(int i = 1, c; i <= K; i++) {
		rf(c)
		D[c] = i;
	}
	for(int i = 1; i <= M; i++) {
		rf(A[i]) rf(B[i]) rf(C[i])
		fg(G, A[i], B[i], C[i]);
	}

	for(int i = 1; i <= N; i++)
		fill(dp[i]+1, dp[i]+(1<<K), INFLL);
	
	for(int i = 1; i <= N; i++) {
		PQ.push(pli(0, i<<5));
		if(D[i]) {
			dp[i][1<<(D[i]-1)] = 0;
			PQ.push(pli(0, (i<<5) | (1<<(D[i]-1))));
		}
	}

	for(; !PQ.empty();) {
		ll dst; int idx, key;
		tie(dst, idx) = PQ.top(); PQ.pop();
		key = idx & 31; idx >>= 5;

		if(dp[idx][key] < dst) continue;

		for(auto &e : G[idx]) {
			int nidx = e.first;
			ll ndst = dst + e.second;

			for(int ki = 0, nkey; ki < (1<<K); ki++) {
				nkey = key | ki;
				ll t = ndst + dp[nidx][ki];
				if(dp[nidx][nkey] <= t) continue;
				dp[nidx][nkey] = t;
				PQ.push(pli(t, (nidx<<5) | nkey));
			}
		}
	}

	for(int i = 1; i <= N; i++)
		upmin(Ans, dp[i][(1<<K)-1]);
	cout << Ans << endl;
	return 0;
}

Compilation message

cities.cpp: In function 'int main()':
cities.cpp:39:7: warning: ignoring return value of 'size_t fread(void*, size_t, size_t, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
  fread(str, 1, 22000022, stdin);
  ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 2684 KB Output is correct
2 Correct 4 ms 2832 KB Output is correct
3 Correct 5 ms 2908 KB Output is correct
4 Correct 4 ms 2956 KB Output is correct
5 Correct 4 ms 2956 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1442 ms 57488 KB Output is correct
2 Correct 1515 ms 57488 KB Output is correct
3 Correct 816 ms 57488 KB Output is correct
4 Correct 118 ms 57488 KB Output is correct
5 Correct 557 ms 57488 KB Output is correct
6 Correct 58 ms 57488 KB Output is correct
7 Correct 9 ms 57488 KB Output is correct
8 Correct 6 ms 57488 KB Output is correct

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 57488 KB Output is correct
2 Correct 16 ms 57488 KB Output is correct
3 Correct 17 ms 57488 KB Output is correct
4 Correct 14 ms 57488 KB Output is correct

Subtask #4
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 4490 ms 106812 KB Output is correct
2 Correct 4284 ms 106812 KB Output is correct
3 Correct 2434 ms 106812 KB Output is correct
4 Correct 3070 ms 106812 KB Output is correct
5 Correct 931 ms 106812 KB Output is correct
6 Correct 323 ms 106812 KB Output is correct

Subtask #5
0 / 26.0

# Verdict Execution time Memory Grader output
1 Execution timed out 6063 ms 172636 KB Time limit exceeded
2 Halted 0 ms 0 KB -