oj.uz

답안 #487359

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
487359 2021-11-15T10:18:59 Z Koosha_mv Cities (BOI16_cities) C++14
51 / 100
 6000 ms 249792 KB 
#include <bits/stdc++.h>
using namespace std;
#define erorp(x) cout<<#x<<"={"<<(x.F)<<" , "<<x.S<<"}"<<endl
#define print(v,r) f(i,0,r) cout<<v[i]<<" "; cout<<endl
#define eror(x) cout<<#x<<'='<<(x)<<endl
#define f_(i,a,b) for(int i=a;i>=b;i--)
#define f(i,a,b) for(int i=a;i<b;i++)
#define nb(x) __builtin_popcount(x)
#define maxm(a,b) a=max(a,b)
#define minm(a,b) a=min(a,b)
#define Add(x,y) x=(x+y)%mod
#define lst(x) x[x.size()-1]
#define sz(x) int(x.size())
#define mp make_pair
#define ll long long
#define pb push_back
#define S second
#define F first

const int N=2e5+99,K=5;
const ll inf=1e15;

int n,k,m,a[N],imp[N],col[N],mark[N][(1<<K)];
ll ans=inf,dp[N][(1<<K)];
vector<pair<int,int> > g[N];
set<pair<ll,pair<int,int> > > s;

void upd(int u,int mask,ll x){
	//cout<<u<<" "<<mask<<" : "<<x<<endl;
	if(dp[u][mask]<=x) return ;
	s.erase(mp(dp[u][mask],mp(u,mask)));
	dp[u][mask]=x;
	s.insert(mp(dp[u][mask],mp(u,mask)));	
}

main(){
	fill(col,col+N,-1);
	cin>>n>>k>>m;
	f(i,0,k){
		cin>>imp[i];
		col[imp[i]]=i;
	}
	f(i,0,m){
		int u,v,c;
		cin>>u>>v>>c;
		g[u].pb(mp(v,c));
		g[v].pb(mp(u,c));
	}
	f(i,1,n+1){
		f(mask,0,(1<<k)){
			dp[i][mask]=inf;
		}
	}
	f(i,1,n+1){
		if(col[i]==-1){
			dp[i][0]=0;
		}
		else{
			dp[i][(1<<col[i])]=0;
		}
	}
	f(i,1,n+1){
		f(mask,0,(1<<k)){
			s.insert(mp(dp[i][mask],mp(i,mask)));
		}
	}
	while(s.size()){
		pair<ll,pair<int,int> > p=*s.begin();
		int u=p.S.F,mask=p.S.S;
		//cout<<u<<" "<<mask<<" : "<<p.F<<endl;
		if(mark[u][mask]){
			cout<<-1;
			exit(0);
		}
		mark[u][mask]=1;
		s.erase(p);
		f(i,0,g[u].size()){
			f(nmask,0,(1<<k)){
				upd(u,mask|nmask,dp[u][mask]+dp[g[u][i].F][nmask]+g[u][i].S);
				upd(g[u][i].F,mask|nmask,dp[u][mask]+dp[g[u][i].F][nmask]+g[u][i].S);
			}
		}
		//cout<<endl;
	}
	f(i,1,n+1){
		minm(ans,dp[i][(1<<k)-1]);
	}
	cout<<ans;
}

Compilation message

cities.cpp:36:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   36 | main(){
      | ^~~~
cities.cpp: In function 'int main()':
cities.cpp:7:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    7 | #define f(i,a,b) for(int i=a;i<b;i++)
......
   77 |   f(i,0,g[u].size()){
      |     ~~~~~~~~~~~~~~~            
cities.cpp:77:3: note: in expansion of macro 'f'
   77 |   f(i,0,g[u].size()){
      |   ^

Subtask #1
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 5708 KB Output is correct
2 Correct 4 ms 5708 KB Output is correct
3 Correct 3 ms 5708 KB Output is correct
4 Correct 3 ms 5708 KB Output is correct
5 Correct 5 ms 5836 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5066 ms 99472 KB Output is correct
2 Correct 4463 ms 99292 KB Output is correct
3 Correct 1670 ms 96572 KB Output is correct
4 Correct 472 ms 10780 KB Output is correct
5 Correct 1564 ms 74424 KB Output is correct
6 Correct 233 ms 10624 KB Output is correct
7 Correct 17 ms 6604 KB Output is correct
8 Correct 8 ms 6476 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 43 ms 7208 KB Output is correct
2 Correct 41 ms 7216 KB Output is correct
3 Correct 21 ms 7116 KB Output is correct
4 Correct 31 ms 6536 KB Output is correct

Subtask #4
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6095 ms 149448 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6103 ms 249792 KB Time limit exceeded
2 Halted 0 ms 0 KB -