标签:NPU strong 优化 name scan IV second lld 结果
lyk拥有一个区间。
Input第一行一个数n(1<=n<=100000)。
接下来一行n个数ai,表示这n个数(0<=ai<=10^9)。Output一行表示答案。Sample Input
3 3 4 5
Sample Output
70
题意:求所有区间的&值乘|值。
思路:分治,利用这两个逻辑运算的收敛性,用map记录相应个数,然后累加即可。
(和前面利用gcd的收敛性分治的一样的。
#include<bits/stdc++.h> #define ll long long using namespace std; const int maxn=100010; const int Mod=1000000007; ll ans; ll a[maxn],And[maxn],Or[maxn]; map<pair<ll,ll>,ll>mp; map<pair<ll,ll>,ll>::iterator it; void solve(int L,int R) { if(L>R) return ; int Mid=(L+R)>>1; mp.clear(); And[Mid]=a[Mid]; Or[Mid]=a[Mid]; mp[make_pair(And[Mid],Or[Mid])]++; for(int i=Mid-1;i>=L;i--){ And[i]=And[i+1]&a[i]; Or[i]=Or[i+1]|a[i]; mp[make_pair(And[i],Or[i])]++; } for(int i=Mid;i<=R;i++){ for(it=mp.begin();it!=mp.end();it++) ans+=(ll)(And[i]&(*it).first.first)*(Or[i]|(*it).first.second)%Mod*(*it).second%Mod; } solve(L,Mid-1); solve(Mid+1,R); } int main() { int N,i,j; scanf("%d",&N); for(i=1;i<=N;i++) scanf("%lld",&a[i]); solve(1,N); printf("%lld\n",ans); return 0; }
小小优化:
#include<bits/stdc++.h> #define ll long long using namespace std; const int maxn=100010; const int Mod=1e9+7; int a[maxn],And[maxn],Or[maxn]; ll ans; map<pair<int,int>,int>mp,tp; map<pair<int,int>,int>::iterator it1,it2; void solve(int L,int R) { if(L>R) return ; int Mid=(L+R)>>1; mp.clear(); tp.clear(); And[Mid]=a[Mid]; Or[Mid]=a[Mid]; mp[make_pair(And[Mid],Or[Mid])]++; for(int i=Mid-1;i>=L;i--){ And[i]=And[i+1]&a[i]; Or[i]=Or[i+1]|a[i]; mp[make_pair(And[i],Or[i])]++; } tp[make_pair(And[Mid],Or[Mid])]++; for(int i=Mid+1;i<=R;i++){ And[i]=And[i-1]&a[i]; Or[i]=Or[i-1]|a[i]; tp[make_pair(And[i],Or[i])]++; } for(it1=mp.begin();it1!=mp.end();it1++) for(it2=tp.begin();it2!=tp.end();it2++) ans=(ans+(ll)((*it1).first.first&(*it2).first.first)*((*it1).first.second|(*it2).first.second)%Mod*(*it1).second*(*it2).second)%Mod; solve(L,Mid-1); solve(Mid+1,R); } int main() { int N,i,j; scanf("%d",&N); for(i=1;i<=N;i++) scanf("%d",&a[i]); solve(1,N); printf("%lld\n",ans); return 0; }
51nod1674:区间的价值2(分治,利用&和|的收敛性)
标签:NPU strong 优化 name scan IV second lld 结果
原文地址:https://www.cnblogs.com/hua-dong/p/9185351.html