#include "wiring.h"
#include <cassert>
#include <cstdio>
#include <vector>
#include <climits>
#include <stdlib.h>
#include <algorithm>
using namespace std;
bool isSubtask1(std::vector<int> r, std::vector<int> b) {return r.size()<=200 && b.size()<=200;}
bool isSubtask2(std::vector<int> r, std::vector<int> b) {return r[r.size()-1]<b[0];}
bool isSubtask3(std::vector<int> r, std::vector<int> b) {
int n=r.size(), m=b.size();
vector<int> a(n+m);
for(int i=0, j=0; i<n || j<m;)
i==n? a[i+j++]=1: j==m? a[j+i++]=0: r[i]<b[j]? a[j+i++]=0:a[i+j++] = 1;
int cnt = 1;
for(int k=1; k<n+m; k++) {
cnt = a[k]==a[k-1]? cnt+1:0;
if (cnt > 6) return false;
}
return true;
}
long long subtask1(std::vector<int>, std::vector<int>);
long long subtask2(std::vector<int> &, std::vector<int> &, int, int, int, int);
long long subtask3(std::vector<int>, std::vector<int>);
long long min_total_length(std::vector<int>, std::vector<int> );
long long min_total_length(std::vector<int> r, std::vector<int> b) {
// return subtask1(r,b);
if (isSubtask3(r,b)) return subtask3(r,b);
else if (isSubtask2(r,b)) return subtask2(r,b,0,r.size()-1,0,b.size()-1);
else if (isSubtask1(r,b)) return subtask1(r,b);
// else if (isSubtask4(r,b)) return subtask4(r,b);
// else return fullTask(r,b);
return 0;
}
long long subtask3(std::vector<int> r, std::vector<int> b) {
int n = r.size(), m = b.size();
vector<pair<int,int>> a(n+m);
for(int i=0, j=0; i<n || j<m;) // make_pair 내부에서 ++ 하면 안됨
i==n? a[i+j++]=make_pair(j,1): j==m? a[j+i++]=make_pair(i,0): r[i]<b[j]? a[j+i++]=make_pair(i,0): a[i+j++]=make_pair(j,1);
// for(int i=0; i<a.size(); i++)
// printf("a[%d] = (%d,%d), ", i,a[i].first,a[i].second);
// printf("\n");
vector<long long> dp(n+m,LLONG_MAX);
dp[0] = 0;
int prev=0, curr=0;
for(curr=1; curr<n+m; curr++)
if (a[curr].second != a[curr-1].second) break;
else dp[curr] = 0;
// printf("prev: %d, curr: %d\n", prev,curr);
for(int i=curr; i<n+m; i++) {
if (a[prev].second==0)
dp[i] = subtask2(r,b,a[prev].first,a[curr-1].first,a[curr].first,a[i].first);
else
dp[i] = subtask2(b,r,a[prev].first,a[curr-1].first,a[curr].first,a[i].first);
// printf("dp[%d] = %lld\n", i,dp[i]);
if(i<n+m-1 && a[i].second != a[i+1].second) {prev=curr; curr=i+1; break;}
}
// printf("prev: %d, curr: %d\n", prev,curr);
// for(int i=0; i<curr; i++)
// printf("dp[%d] = %lld\n", i,dp[i]);
bool start = true;
for(int i=curr; i<n+m; i++) {
// printf("prev: %d, curr: %d\n", prev,curr);
for(int j =prev; j<curr; j++) {
// printf("i:%d, j:%d\n",i,j);
long long temp;
if (a[prev].second==0)
temp = subtask2(r,b,a[j].first,a[curr-1].first,a[curr].first,a[i].first);
else
temp = subtask2(b,r,a[j].first,a[curr-1].first,a[curr].first,a[i].first);
// printf("temp=%d ", temp);
if (j==prev && prev == curr-1) {temp += min(dp[j-1],dp[j]);}
else temp += dp[j-1];
if (dp[i]>temp) dp[i] = temp;
// printf("dp[%d] = %lld\n", i,dp[i]);
}
if (i<n+m-1 && a[i].second != a[i+1].second) {prev = curr; curr = i+1;}
}
return dp[n+m-1];
}
long long subtask2(std::vector<int> &arr1, std::vector<int> &arr2, int s1, int e1, int s2, int e2) {
long long sum1=0, sum2=0, n = e1-s1+1, m=e2-s2+1;
for(int i=s1; i<=e1; i++) sum1 += arr1[e1] - arr1[i];
for(int i=s2; i<=e2; i++) sum2 += arr2[i] - arr2[s2];
return sum1+sum2+max(n,m)*(arr2[s2]-arr1[e1]);
}
long long subtask1(std::vector<int> r, std::vector<int> b) {
int n = r.size(), m = b.size();
long long dp[n][m];
dp[0][0] = abs(r[0]-b[0]);
for(int i=1; i<n; i++) dp[i][0] = dp[i-1][0] + abs(r[i]-b[0]);
for(int j=1; j<m; j++) dp[0][j] = dp[0][j-1] + abs(r[0]-b[j]);
for(int i=1; i<n; i++)
for(int j=1; j<m; j++)
dp[i][j] = min(min(dp[i-1][j],dp[i][j-1]),dp[i-1][j-1]) + abs(r[i]-b[j]);
return dp[n-1][m-1];
}
Compilation message
wiring.cpp: In function 'long long int subtask3(std::vector<int>, std::vector<int>)':
wiring.cpp:75:10: warning: unused variable 'start' [-Wunused-variable]
75 | bool start = true;
| ^~~~~
wiring.cpp: In function 'long long int subtask1(std::vector<int>, std::vector<int>)':
wiring.cpp:111:7: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
111 | for(int j=1; j<m; j++)
| ^~~
wiring.cpp:114:2: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
114 | return dp[n-1][m-1];
| ^~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
404 KB |
Output is correct |
3 |
Correct |
1 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
344 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
604 KB |
Output is correct |
8 |
Correct |
1 ms |
604 KB |
Output is correct |
9 |
Correct |
0 ms |
604 KB |
Output is correct |
10 |
Correct |
1 ms |
600 KB |
Output is correct |
11 |
Correct |
1 ms |
604 KB |
Output is correct |
12 |
Correct |
0 ms |
604 KB |
Output is correct |
13 |
Correct |
0 ms |
604 KB |
Output is correct |
14 |
Correct |
1 ms |
604 KB |
Output is correct |
15 |
Correct |
1 ms |
604 KB |
Output is correct |
16 |
Correct |
1 ms |
604 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
13 ms |
2652 KB |
Output is correct |
4 |
Correct |
13 ms |
2652 KB |
Output is correct |
5 |
Correct |
14 ms |
2556 KB |
Output is correct |
6 |
Correct |
18 ms |
3420 KB |
Output is correct |
7 |
Correct |
17 ms |
3516 KB |
Output is correct |
8 |
Correct |
17 ms |
3528 KB |
Output is correct |
9 |
Correct |
18 ms |
3420 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
3rd lines differ - on the 1st token, expected: '316', found: '188' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
3rd lines differ - on the 1st token, expected: '27', found: '23' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
404 KB |
Output is correct |
3 |
Correct |
1 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
344 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
604 KB |
Output is correct |
8 |
Correct |
1 ms |
604 KB |
Output is correct |
9 |
Correct |
0 ms |
604 KB |
Output is correct |
10 |
Correct |
1 ms |
600 KB |
Output is correct |
11 |
Correct |
1 ms |
604 KB |
Output is correct |
12 |
Correct |
0 ms |
604 KB |
Output is correct |
13 |
Correct |
0 ms |
604 KB |
Output is correct |
14 |
Correct |
1 ms |
604 KB |
Output is correct |
15 |
Correct |
1 ms |
604 KB |
Output is correct |
16 |
Correct |
1 ms |
604 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
0 ms |
348 KB |
Output is correct |
19 |
Correct |
13 ms |
2652 KB |
Output is correct |
20 |
Correct |
13 ms |
2652 KB |
Output is correct |
21 |
Correct |
14 ms |
2556 KB |
Output is correct |
22 |
Correct |
18 ms |
3420 KB |
Output is correct |
23 |
Correct |
17 ms |
3516 KB |
Output is correct |
24 |
Correct |
17 ms |
3528 KB |
Output is correct |
25 |
Correct |
18 ms |
3420 KB |
Output is correct |
26 |
Incorrect |
0 ms |
348 KB |
3rd lines differ - on the 1st token, expected: '316', found: '188' |
27 |
Halted |
0 ms |
0 KB |
- |