POJ 3468 (线段树 区间增减) A Simple Problem with Integers

这题WA了好久，一直以为是lld和I64d的问题，后来发现是自己的pushdown函数写错了，说到底还是因为自己对线段树理解得不好。

因为是懒惰标记，所以只有在区间分开的时候才会将标记往下传递。更新和查询都要pushdown。

 1 #include <cstdio>

 2 

 3 typedef long long LL;

 4 

 5 const int maxn = 100000 + 10;

 6 

 7 int n, m, qL, qR, v;

 8 LL sum[maxn << 2], add[maxn << 2];

 9 

10 inline void maintain(int o)

11 { sum[o] = sum[o*2] + sum[o*2+1]; }

12 

13 void build(int o, int L, int R)

14 {

15     if(L == R) { scanf("%I64d", &sum[o]); return; }

16     int M = (L + R) / 2;

17     build(o*2, L, M);

18     build(o*2+1, M+1, R);

19     maintain(o);

20 }

21 

22 void pushdown(int o, int L, int R)

23 {

24     if(add[o])

25     {

26         int lc = o*2, rc = o*2+1;

27         int M = (L + R) / 2;

28         add[lc] += add[o];

29         add[rc] += add[o];

30         sum[lc] += (LL) add[o] * (M - L + 1);

31         sum[rc] += (LL) add[o] * (R - M);

32         add[o] = 0;

33     }

34 }

35 

36 void update(int o, int L, int R)

37 {

38     if(qL <= L && qR >= R)

39     {

40         add[o] += v;

41         sum[o] += (LL) v * (R - L + 1);

42         return;

43     }

44     int M = (L + R) / 2;

45     pushdown(o, L, R);

46     if(qL <= M) update(o*2, L, M);

47     if(qR > M) update(o*2+1, M+1, R);

48     maintain(o);

49 }

50 

51 LL query(int o, int L, int R)

52 {

53     if(qL <= L && qR >= R) return sum[o];

54     pushdown(o, L, R);

55     int M = (L + R) / 2;

56     LL ans = 0;

57     if(qL <= M) ans += query(o*2, L, M);

58     if(qR > M) ans += query(o*2+1, M+1, R);

59     return ans;

60 }

61 

62 int main()

63 {

64     //freopen("in.txt", "r", stdin);

65 

66     scanf("%d%d", &n, &m);

67     build(1, 1, n);

68     char op[10];

69     while(m--)

70     {

71         scanf("%s", op);

72         if(op[0] == 'Q')

73         {

74             scanf("%d%d", &qL, &qR);

75             printf("%I64d\n", query(1, 1, n));

76         }

77         else

78         {

79             scanf("%d%d%d", &qL, &qR, &v);

80             update(1, 1, n);

81         }

82     }

83 

84     return 0;

85 }

代码君