比赛 EYOI与SBOI开学欢乐赛12th 评测结果 AAAATTTTTT
题目名称 坐标压缩 最终得分 40
用户昵称 yrtiop 运行时间 12.104 s
代码语言 C++ 内存使用 29.77 MiB
提交时间 2022-10-17 21:22:23
显示代码纯文本
  1. #include <bits/stdc++.h>
  2. #define pb emplace_back
  3.  
  4. const int maxn = 3e5 + 5;
  5. int n,a[maxn],s,c[maxn],cnt,pos[maxn];
  6. std::vector<int> G[maxn];
  7.  
  8. int ls[maxn << 2],rs[maxn << 2],sum[maxn << 2];
  9.  
  10. void pushup(int i) {
  11. sum[i] = std::max(sum[i << 1] , sum[i << 1 | 1]);
  12. return ;
  13. }
  14.  
  15. void build(int i,int l,int r) {
  16. ls[i] = l;
  17. rs[i] = r;
  18. sum[i] = 0;
  19. if(l == r)return ;
  20. int mid = l + r >> 1;
  21. build(i << 1 , l , mid);
  22. build(i << 1 | 1 , mid + 1 , r);
  23. return ;
  24. }
  25.  
  26. void modify(int i,int pos,int val) {
  27. if(ls[i] == rs[i]) {
  28. sum[i] = val;
  29. return ;
  30. }
  31. int mid = ls[i] + rs[i] >> 1;
  32. if(pos <= mid)modify(i << 1 , pos , val);
  33. else modify(i << 1 | 1 , pos , val);
  34. pushup(i);
  35. return ;
  36. }
  37.  
  38. int query(int i,int l,int r) {
  39. if(ls[i] >= l&&rs[i] <= r)return sum[i];
  40. if(ls[i] > r||rs[i] < l)return 0;
  41. int mid = ls[i] + rs[i] >> 1,s = 0;
  42. if(l <= mid)s = std::max(s , query(i << 1 , l , r));
  43. if(r > mid)s = std::max(s , query(i << 1 | 1 , l , r));
  44. return s;
  45. }
  46.  
  47. bool check(int k) {
  48. int ans = 0;
  49. build(1 , 1 , n);
  50. for(int i = 1;i <= cnt;++ i) {
  51. for(auto& v : G[i]) {
  52. int l = std::max(v - k , 1),r = std::min(v + k , n);
  53. pos[v] = query(1 , l , r) + 1;
  54. ans += pos[v];
  55. if(ans > s)return false;
  56. }
  57. for(auto& v : G[i]) {
  58. modify(1 , v , pos[v]);
  59. }
  60. }
  61. return ans <= s;
  62. }
  63.  
  64. void work() {
  65. scanf("%d %d",&n,&s);
  66. cnt = 0;
  67. for(int i = 1;i <= n;++ i)
  68. scanf("%d",&a[i]),c[++ cnt] = a[i];
  69. std::sort(c + 1 , c + 1 + cnt);
  70. cnt = std::unique(c + 1 , c + 1 + cnt) - c - 1;
  71. for(int i = 1;i <= n;++ i)
  72. a[i] = std::lower_bound(c + 1 , c + 1 + cnt , a[i]) - c;
  73. for(int i = 1;i <= cnt;++ i)
  74. G[i].clear();
  75. for(int i = 1;i <= n;++ i)
  76. G[a[i]].pb(i);
  77. int l = 0,r = n;
  78. while(l <= r) {
  79. int mid = l + r >> 1;
  80. if(check(mid))l = mid + 1;
  81. else r = mid - 1;
  82. }
  83. printf("%d\n",r + 1);
  84. return ;
  85. }
  86.  
  87. int main() {
  88. freopen("coord.in","r",stdin);
  89. freopen("coord.out","w",stdout);
  90. int T;
  91. scanf("%d",&T);
  92. while(T --)work();
  93. return 0;
  94. }