首页 > 代码库 > POJ3468A Simple Problem with Integers
POJ3468A Simple Problem with Integers
#include<iostream> #include<cstdio> #include<algorithm> using namespace std; struct CNode { int L,R; CNode* pLeft,* pRight; long long Inc; long long nSum; }; CNode Tree[200010]; int nCount=0; int Mid(CNode* pRoot) { return (pRoot->L+pRoot->R)/2; } void BuildTree(CNode* pRoot,int L,int R) { pRoot->L=L; pRoot->R=R; pRoot->nSum=0; pRoot->Inc=0; if(L==R) return; nCount++; pRoot->pLeft=Tree+nCount; nCount++; pRoot->pRight=Tree+nCount; BuildTree(pRoot->pLeft,L,(L+R)/2); BuildTree(pRoot->pRight,(L+R)/2+1,R); } void Insert(CNode* pRoot,int i,int v) { if(pRoot->L==i&&pRoot->R==i) { pRoot->nSum=v; return; } pRoot->nSum+=v; if(i<=Mid(pRoot)) { Insert(pRoot->pLeft,i,v); } else { Insert(pRoot->pRight,i,v); } } void Add(CNode* pRoot,int a,int b,long long c) { if(pRoot->L==a&&pRoot->R==b) { pRoot->Inc+=c; return; } pRoot->nSum+=c*(b-a+1); if(b<=Mid(pRoot)) { Add(pRoot->pLeft,a,b,c); } else if(a>Mid(pRoot)) { Add(pRoot->pRight,a,b,c); } else { Add(pRoot->pLeft,a,Mid(pRoot),c); Add(pRoot->pRight,Mid(pRoot)+1,b,c); } } long long QuerynSum(CNode* pRoot,int a,int b) { if(pRoot->L==a&&pRoot->R==b) { return pRoot->nSum+(pRoot->R-pRoot->L+1)*pRoot->Inc; } pRoot->nSum+=(pRoot->R-pRoot->L+1)*pRoot->Inc; Add(pRoot->pLeft,pRoot->L,Mid(pRoot),pRoot->Inc); Add(pRoot->pRight,Mid(pRoot)+1,pRoot->R,pRoot->Inc); pRoot->Inc=0; if(b<=Mid(pRoot)) { return QuerynSum(pRoot->pLeft,a,b); } else if(a>Mid(pRoot)) { return QuerynSum(pRoot->pRight,a,b); } else { return QuerynSum(pRoot->pLeft,a,Mid(pRoot))+QuerynSum(pRoot->pRight,Mid(pRoot)+1,b); } } int main() { int n,q,a,b,c; char cmd[10]; //freopen("d:\\test.txt","r",stdin); scanf("%d%d",&n,&q); int i,j,k; nCount= 0; BuildTree(Tree,1,n); for( i= 1; i <= n; i++ ) { scanf("%d",&a); Insert(Tree,i,a); } for( i= 0; i < q; i++ ) { scanf("%s",cmd); if ( cmd[0] == 'C' ) { scanf("%d%d%d",&a,&b,&c); Add( Tree,a,b,c); } else { scanf("%d%d",&a,&b); printf("%I64d\n",QuerynSum(Tree,a,b)); } } return 0; }
POJ3468A Simple Problem with Integers
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。