树状数组
题意:有两排城市,这两排之间有一些城市之间有连接的道路,给出所有道路,问有多少道路是相交的。
分析:求逆序数。我们先把所有的道路按照a升序,a相同时b升序的方法排列。这样从头至尾便利,对于每条道路,我们只需要知道它之前有多少道路的b大于它的b就可以了,所以我们只要知道前面有多少b小于等于它的再用下标减去就可以了。而这个求有多少小于等于的过程就用树状数组来实现。我们每看到一条边,就把它的b作为下标,把树状数组对应位进行修改。这样一来,树状数组所有下标小于等于该道路的b的数的总和就是我们要求的b小于等于该道路的道路数。
开始我看了一个程序,是按照b降序,b相同时a升序。每次找a比它小的,这样很难处理b相等的情况,所以不推荐。
#include#include #include #include #include using namespace std;#define maxn 2005struct Highway{ int a, b;} way[maxn * maxn];int n, m, k, c[maxn];long long ans;void input(){ scanf("%d%d%d", &n, &m, &k); for (int i = 0; i < k; i++) scanf("%d%d", &way[i].a, &way[i].b);}bool cmp(const Highway &a, const Highway &b){ if (a.a != b.a) return a.a < b.a; return a.b < b.b;}long long cal(int a){ long long sum = 0; while (a > 0) { sum += c[a]; a -= (a & -a); } return sum;}void modify(int a, int x){ while (a <= m) { c[a] += x; a += (a & -a); }}void work(){ ans = 0; for (int i = 0; i < k; i++) { ans += i - cal(way[i].b); modify(way[i].b, 1); }}int main(){ //freopen("D:\\t.txt", "r", stdin); int t; scanf("%d", &t); for (int i = 0; i < t; i++) { memset(c, 0, sizeof(c)); input(); sort(way, way + k, cmp); work(); printf("Test case %d: %I64d\n", i + 1, ans); } return 0;}