LightOJ 1372 （枚举 + 树状数组）

题目

Link
输出序列中有多少个组合 {a1,a2,a3,a4,a5,a6}可以构成一个六边形。

分析

序列每个数都不相等。
所以可以设 a1<a2<a3<a4<a5<a6。
把六边形分解为 4 个三角形， 又可以得出
a1+a2+a3+a4+a5>a6:
a1+a2+a3>a6−a5−a4。
在 a4 固定的情况下， a3可以取[a3,a4)之间
所以我们枚举 ， 用树状数组维护。

Code

#include <bits/stdc++.h>using namespace std;const int maxn = 5000000 + 131;int C[maxn];int lowbit(int x) {    return x &(-x);}int Sum(int x) {    int ret = 0;    while(x) {        ret += C[x];        x -= lowbit(x);    }    return ret;}void Add(int pos) {    while(pos < maxn) {        C[pos] ++;        pos += lowbit(pos);    }}int Num[105];int main() {    int T;    scanf("%d",&T);    for(int kase = 1; kase <= T; ++kase) {        memset(C, 0, sizeof(C));        int N;        scanf("%d",&N);        for(int i = 0; i < N; ++i)            scanf("%d",Num+i);        sort(Num, Num+N);        int Ans = 0, LowZ = 0;     //小于0的情况用LowZ 统计        for(int i = N-1; i >= 0; --i) {            for(int j = 0; j < i; ++j)            for(int k = j+1; k < i; ++k)            {                int sum = Num[i] + Num[j] + Num[k];                Ans += Sum(sum-1);                Ans += LowZ;            }            for(int j = i + 1; j < N; ++j)            for(int k = j + 1; k < N; ++k)            {                int sub = Num[k] - Num[j] - Num[i];                if(sub > 0) Add(sub);                else LowZ ++;            }        }        printf("Case %d: %d\n",kase, Ans);    }    return 0;}