Submission #304437

#TimeUsernameProblemLanguageResultExecution timeMemory
304437urd05Carnival Tickets (IOI20_tickets)C++14
Compilation error
0 ms0 KiB
long long dp[81][6400];
const long long INF=1e18;
int zero[80];
int one[80];
typedef pair<int,int> P;
 
long long find_maximum(int k, vector<vector<int>> v) {
	int n = v.size();
	int m = v[0].size();
	vector<vector<int>> answer(n,vector<int>(m));
	for(int i=0;i<n;i++) {
	    for(int j=0;j<m;j++) {
	        answer[i][j]=-1;
	    }
	}
	dp[0][0]=0;
	for(int i=1;i<=(n*k)/2;i++) {
	    dp[0][i]=-INF;
	}
	for(int i=1;i<=n;i++) {
	    for(int j=0;j<=(n*k)/2;j++) {
	        dp[i][j]=-INF;
	        long long sum=0;
	        for(int l=0;l<k;l++) {
	            sum-=v[i-1][l];
	        }
	        for(int l=0;l<=k&&l<=j;l++) {
	            dp[i][j]=max(dp[i][j],dp[i-1][j-l]+sum);
	            if (l!=k) {
	                sum+=v[i-1][m-l-1];
	                sum+=v[i-1][k-l-1];
	            }
	        }
	    }
	}
	long long ret=dp[n][(n*k)/2];
	int now=(n*k)/2;
	for(int i=n-1;i>=0;i--) {
	    int val;
	    long long sum=0;
	    for(int j=0;j<k;j++) {
	        sum-=v[i][j];
	    }
	    for(int j=0;j<=k&&j<=now;j++) {
	        if (dp[i][now-j]+sum==dp[i+1][now]) {
	            val=j;
	            now-=j;
	            break;
	        }
	        sum+=v[i][m-j-1];
	        sum+=v[i][k-j-1];
	    }
	    zero[i]=k-val;
	    one[i]=val;
	}
	for(int cnt=0;cnt<k;cnt++) {
	    vector<int> ind;
	    priority_queue<P> pq;
	    int pick=n/2;
	    for(int i=0;i<n;i++) {
	        if (zero[i]==0) {
	            ind.push_back(i);
	            pick--;
	        }
	        if (one[i]==0) {
	            continue;
	        }
	        pq.push(P(one[i]-zero[i],i));
	    }
	    for(int i=0;i<pick;i++) {
	        ind.push_back(pq.top().second);
	        pq.pop();
	    }
	    sort(ind.begin(),ind.end());
	    for(int i=0;i<n;i++) {
	        if (binary_search(ind.begin(),ind.end(),i)) {
	            answer[i][m-one[i]]=cnt;
	            one[i]--;
	        }
	        else {
	            answer[i][zero[i]-1]=cnt;
	            zero[i]--;
	        }
	    }
	}
	allocate_tickets(answer);
	return ret;
}

Compilation message (stderr)

tickets.cpp:5:9: error: 'pair' does not name a type
    5 | typedef pair<int,int> P;
      |         ^~~~
tickets.cpp:7:31: error: 'vector' has not been declared
    7 | long long find_maximum(int k, vector<vector<int>> v) {
      |                               ^~~~~~
tickets.cpp:7:37: error: expected ',' or '...' before '<' token
    7 | long long find_maximum(int k, vector<vector<int>> v) {
      |                                     ^
tickets.cpp: In function 'long long int find_maximum(int, int)':
tickets.cpp:8:10: error: 'v' was not declared in this scope
    8 |  int n = v.size();
      |          ^
tickets.cpp:10:2: error: 'vector' was not declared in this scope
   10 |  vector<vector<int>> answer(n,vector<int>(m));
      |  ^~~~~~
tickets.cpp:10:16: error: expected primary-expression before 'int'
   10 |  vector<vector<int>> answer(n,vector<int>(m));
      |                ^~~
tickets.cpp:13:10: error: 'answer' was not declared in this scope
   13 |          answer[i][j]=-1;
      |          ^~~~~~
tickets.cpp:28:23: error: 'max' was not declared in this scope
   28 |              dp[i][j]=max(dp[i][j],dp[i-1][j-l]+sum);
      |                       ^~~
tickets.cpp:57:13: error: expected primary-expression before 'int'
   57 |      vector<int> ind;
      |             ^~~
tickets.cpp:58:6: error: 'priority_queue' was not declared in this scope
   58 |      priority_queue<P> pq;
      |      ^~~~~~~~~~~~~~
tickets.cpp:58:21: error: 'P' was not declared in this scope
   58 |      priority_queue<P> pq;
      |                     ^
tickets.cpp:58:24: error: 'pq' was not declared in this scope
   58 |      priority_queue<P> pq;
      |                        ^~
tickets.cpp:62:14: error: 'ind' was not declared in this scope; did you mean 'int'?
   62 |              ind.push_back(i);
      |              ^~~
      |              int
tickets.cpp:71:10: error: 'ind' was not declared in this scope; did you mean 'int'?
   71 |          ind.push_back(pq.top().second);
      |          ^~~
      |          int
tickets.cpp:74:11: error: 'ind' was not declared in this scope; did you mean 'int'?
   74 |      sort(ind.begin(),ind.end());
      |           ^~~
      |           int
tickets.cpp:74:6: error: 'sort' was not declared in this scope; did you mean 'short'?
   74 |      sort(ind.begin(),ind.end());
      |      ^~~~
      |      short
tickets.cpp:76:14: error: 'binary_search' was not declared in this scope
   76 |          if (binary_search(ind.begin(),ind.end(),i)) {
      |              ^~~~~~~~~~~~~
tickets.cpp:77:14: error: 'answer' was not declared in this scope
   77 |              answer[i][m-one[i]]=cnt;
      |              ^~~~~~
tickets.cpp:81:14: error: 'answer' was not declared in this scope
   81 |              answer[i][zero[i]-1]=cnt;
      |              ^~~~~~
tickets.cpp:86:19: error: 'answer' was not declared in this scope
   86 |  allocate_tickets(answer);
      |                   ^~~~~~
tickets.cpp:86:2: error: 'allocate_tickets' was not declared in this scope
   86 |  allocate_tickets(answer);
      |  ^~~~~~~~~~~~~~~~