POJ 3067 Japan 树状数组 .

原创于 2016-08-29 14:40:30 发布 · 403 阅读

0 ·

CC 4.0 BY-SA版权

POJ+百练同时被 2 个专栏收录

120 篇文章

订阅专栏

数据结构：树状数组

5 篇文章

订阅专栏

本文提供了一种使用线段树解决POJ 3067问题的有效方法。通过初始化线段树并实现求和及更新操作，代码能够高效地处理边权更新和区间查询。适用于对数据结构和算法有兴趣的学习者。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：http://poj.org/problem?id=3067

要用long long类型保存答案，不然WA

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
typedef pair<int,int> Edge;
typedef long long LL;
const int maxn=1000+5;
int lowbit[maxn],C[maxn];
vector<Edge> edges;
void init(int n)
{
	for(int i=1;i<=n;i++)
	lowbit[i]=i&(-i);
}
int Sum(int p) //a[1]+a[2]+....+a[p] 
{
	int sum=0;
	while(p>0){
		sum+=C[p];
		p-=lowbit[p];
	}
	return sum;
}
void Modify(int p,int val,int n)
{
	while(p<=n){
		C[p]+=val;
		p+=lowbit[p];
	}
}
int main()
{
	int T,kase=0;
	init(maxn);
	cin>>T;
	while(T--)
	{
		int n,m,k;
		scanf("%d%d%d",&n,&m,&k);
		edges.clear();
		while(k--)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			edges.push_back(Edge(u,v));
		}
		sort(edges.begin(),edges.end());
		LL sum=0;
		memset(C,0,sizeof(C));
		for(int i=0;i<edges.size();i++)
		{
			int u=edges[i].first,v=edges[i].second;
			sum+=i-Sum(v);
			Modify(v,1,m);
		}
		printf("Test case %d: %lld\n",++kase,sum);
	}
	return 0;
}