当前位置:Gxlcms > html代码 > CodeforcesRound#220(Div.2)D树状数组&&二分_html/css_WEB-ITnose

CodeforcesRound#220(Div.2)D树状数组&&二分_html/css_WEB-ITnose

时间:2021-07-01 10:21:17 帮助过:9人阅读

/*题目*/


题意:给了n,m,然后一个包含m个数的数组nnum,数组默认从小到大排序,然后是 n个操作,输入一个数x,若x为0,把0加到这个字符串的末尾,若x为1,把1加到这个字符串的末尾,若x为-1,那么把字符串里的 下标 与 nnum数组里的元素相等的 给删除,字符串一开始是空的,问你最后字符串里有什么,若为空 就输出 POOR STACK

这题目看这操作一般都很容易联想到线段树,树状数组,一开始我建了个树状数组,但是超时了,毕竟操作很多,而且 在删除操作里,若nnum数组很大,等同于10^12的复杂度,只能优化,一般来说是用二分了,直接以这个串的某个位置是否为空来建立树状数组,每一次加在最后就不用说了,直接加就可以,关键在于删除,首先删除,要第一步二分查找出 要删除的最后一位,比如字符串长度为7,而nnum数组里 有个8,其实8这个位置不需要删除的,但是这个还是超时,于是想到在树状数组里删除的时候也需要二分,可是写了一直错,后来借鉴了杰哥的发现,原来WA在这里,比如 当前字符串为01010,需要删除 1,3,4位置的字符,要轮着来,当你删除1号位置的时候,字符串变成了1010,你在树状数组里也是要删除1节点的状态,所以原来的3号位置在当前变成了2号位置,当你删了3号以后变成了110 ,原来的4号位置变成了2号位置,所以每删除一次,下标会发生变化,但是没事,发现前面操作了几次而你原来的下标就比当前 减少了几,所以 nnum[i] - i 其实就是 当前需要删除的 元素 在 树状数组里的位置


  1. int n,m;int c[1000000 + 55];int nnum[1000000 + 55];int aa[1000000 + 55];int sum[1000000 + 55];int len;void init() {
  2. memset(c,0,sizeof(c));
  3. memset(aa,-1,sizeof(aa));}bool input() {
  4. while(cin>>n>>m) {
  5. for(int i=0;i<m;i++)cin>>nnum[i];
  6. return false;
  7. }
  8. return true;}int lowbit(int x) {
  9. return x&(-x);}void add(int i,int val) {
  10. while(i <= n) {
  11. c[i] += val;
  12. i += lowbit(i);
  13. }}int get_sum(int i) {
  14. int sum = 0;
  15. while(i > 0) {
  16. sum += c[i];
  17. i -= lowbit(i);
  18. }
  19. return sum;}void binary_find(int pos,int id) {
  20. int l = 1;
  21. int r = n;
  22. while(l <= r) {
  23. int mid = (l + r)>>1;
  24. int ret = get_sum(mid);
  25. if(aa[mid] == -1) {
  26. if(ret >= nnum[pos] - id) r = mid - 1;
  27. else l = mid + 1;
  28. }
  29. else if(ret == nnum[pos] - id) {
  30. add(mid,-1);
  31. aa[mid] = -1;
  32. len--;
  33. break;
  34. }
  35. else if(ret > nnum[pos] - id) r = mid - 1;
  36. else if(ret < nnum[pos] - id) l = mid + 1;
  37. }}void cal() {
  38. int q = n;
  39. len = 0;
  40. int cnt = 1;
  41. while(q--) {
  42. int type;
  43. cin>>type;
  44. if(type == -1) {
  45. if(len == 0)continue;
  46. if(nnum[0] > len)continue;
  47. int now = lower_bound(nnum,nnum + m,len) - nnum;
  48. for(int i=0;i<=now;i++)
  49. binary_find(i,i);
  50. }
  51. else {
  52. aa[cnt] = type;
  53. add(cnt,1);
  54. cnt++;
  55. len++;
  56. }
  57. }
  58. if(len <= 0){puts("Poor stack!");return ;}
  59. for(int i=1;i<=n;i++)
  60. if(aa[i] != -1)
  61. printf("%d",aa[i]);
  62. puts("");}void output() {}int main() {
  63. while(true) {
  64. init();
  65. if(input())return 0;
  66. cal();
  67. output();
  68. }
  69. return 0;}</m;i++)cin>

人气教程排行