比赛 2024.5.23练习赛 评测结果 WAAAA
题目名称 不重叠正方形 最终得分 80
用户昵称 zxhhh 运行时间 1.617 s
代码语言 C++ 内存使用 40.45 MiB
提交时间 2024-05-23 19:35:06
显示代码纯文本
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4. const int N = 1005;
  5. typedef long long ll;
  6. int n, m, a[N][N];
  7. ll s[N][N], v[N][N], f[N][4], ans, g[N][N][2], p[N];
  8.  
  9. void solve (int op) {
  10. memset(f, -0x3f, sizeof f); memset(g, -0x3f, sizeof g);
  11. for (int i = 1;i <= n;i++) {
  12. for (int j = 1;j <= n;j++) {
  13. s[i][j] = s[i-1][j]+s[i][j-1]-s[i-1][j-1];
  14. if (op == 0) s[i][j] += a[i][j];
  15. if (op == 1) s[i][j] += a[n-i+1][n-j+1];
  16. if (op == 2) s[i][j] += a[n-i+1][j];
  17. if (op == 3) s[i][j] += a[i][n-j+1];
  18. if (i >= m && j >= m) v[i][j] = s[i][j]-s[i-m][j]-s[i][j-m]+s[i-m][j-m];
  19. else v[i][j] = -1e18;
  20. }
  21. }
  22. for (int i = n;i >= 1;i--) {
  23. for (int j = 1;j <= n;j++) {
  24. g[i][j][0] = max({v[i][j], g[i+1][j][0], g[i][j-1][0]});
  25. }
  26. for (int j = n;j >= 1;j--) {
  27. g[i][j][1] = max({v[i][j], g[i+1][j][1], g[i][j+1][1]});
  28. }
  29. }
  30. for (int i = 1;i <= n;i++) f[i][0] = 0;
  31. for (int i = m;i <= n;i++) {
  32. ll k = 0;
  33. for (int j = m;j <= n;j++) {
  34. k = max(k, v[i][j]);
  35. }
  36. for (int j = 1;j <= 3;j++) f[i][j] = max(f[i-1][j], f[i-m][j-1]+k);
  37. }
  38. ans = max(ans, f[n][3]);
  39. for (int i = m;i <= n;i++) {
  40. p[i] = p[i-1];
  41. for (int j = m;j <= n;j++) {
  42. p[i] = max(p[i], v[i][j]);
  43. }
  44. for (int j = m;j <= n;j++) {
  45. ll k = v[i][j]+p[i-m]+max(j-m >= m ? g[i][j-m][0] : -1e18, j+m <= n ? g[i][j+m][1] : -1e18);
  46. ans = max(ans, k);
  47. }
  48. }
  49. }
  50.  
  51. int main () {
  52. freopen("zfx.in", "r", stdin);
  53. freopen("zfx.out", "w", stdout);
  54. cin >> n >> m;
  55. for (int i = 1;i <= n;i++) {
  56. for (int j = 1;j <= n;j++) {
  57. cin >> a[i][j];
  58. }
  59. }
  60. solve(0);
  61. solve(1);
  62. solve(2);
  63. solve(3);
  64. cout << ans;
  65. return 0;
  66. }