CC countari & 分块+FFT
题意:
求一个序列中顺序的长度为3的等差数列.
SOL:
对于这种计数问题都是用个数的卷积来进行统计.然而对于这个题有顺序的限制,不好直接统计,于是竟然可以分块?惊为天人...
考虑分块以后的序列:
一个块内直接枚举统计三个或两个在块内的.
只有一个在当前块我们假设它是中间那个,对左右其它块做卷积.
但是还是感觉复杂度有点玄学啊...
我比较傻逼...一开始块内统计根本没有想清楚...最后做卷积硬生生把复杂度变成了 $\sqrt{N}*N*log(N)$...
改了一个晚上终于没忍住看标程...我是傻逼明明a的范围比n小...
CODE:(tle的代码不想改了= =)
/*=================================
# Created time: 2016-04-19 16:09
# Filename: cccountari.cpp
# Description: Ploblem from codechef countari
=================================*/
#define me AcrossTheSky&HalfSummer11
#include <cstdio>
#include <cmath>
#include <ctime>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm> #include <set>
#include <stack>
#include <queue>
#include <vector> #define lowbit(x) (x)&(-x)
#define Abs(x) ((x) > 0 ? (x) : (-(x)))
#define FOR(i,a,b) for((i)=(a);(i)<=(b);(i)++)
#define FORP(i,a,b) for(int i=(a);i<=(b);i++)
#define FORM(i,a,b) for(int i=(a);i>=(b);i--)
#define ls(a,b) (((a)+(b)) << 1)
#define rs(a,b) (((a)+(b)) >> 1)
#define getlc(a) ch[(a)][0]
#define getrc(a) ch[(a)][1] #define maxn 500005
#define maxm 100005
#define INF 1070000000
using namespace std;
typedef long long ll;
typedef unsigned long long ull; template<class T> inline
void read(T& num){
num = 0; bool f = true;char ch = getchar();
while(ch < '0' || ch > '9') { if(ch == '-') f = false;ch = getchar();}
while(ch >= '0' && ch <= '9') {num = num * 10 + ch - '0';ch = getchar();}
num = f ? num: -num;
}
int outs[100];
template<class T> inline
void write(T x){
if (x==0) {putchar('0'); putchar(' '); return;}
if (x<0) {putchar('-'); x=-x;}
int num=0;
while (x){ outs[num++]=(x%10); x=x/10;}
FORM(i,num-1,0) putchar(outs[i]+'0'); putchar(' ');
}
/*==================split line==================*/
const double pi=acos(-1);
struct cpx{
double x,y;
inline double real(){return x;}
cpx(double a=0,double b=0):x(a),y(b){}
}f[maxn],g[maxn],eps[maxn],inv_eps[maxn]; inline cpx operator +(cpx a,cpx b){return cpx(a.x+b.x,a.y+b.y);}
inline cpx operator -(cpx a,cpx b){return cpx(a.x-b.x,a.y-b.y);}
inline cpx operator /(cpx a,double b){return cpx(a.x/b,a.y);}
inline cpx operator *(cpx a,cpx b){return cpx(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
inline cpx conj(cpx a){return cpx(a.x,-a.y);} inline void get_eps(int p){
double angle=2.0*pi/p;
FORP(i,0,p-1) eps[i]=cpx(cos(angle*i),sin(angle*i));
FORP(i,0,p-1) inv_eps[i]=conj(eps[i]);
}
inline void fft(int n,cpx* buffer,cpx* eps){
for (int i = 0, j = 0;i < n; i++){
if (i > j) swap(buffer[i],buffer[j]);
for (int l = n >> 1;(j ^= l) < l;l >>= 1);
}
for (int i = 2; i <= n; i <<= 1){
int m = i >> 1;
for (int j = 0; j < n; j += i)
for (int k = 0; k != m; k++){
cpx z = buffer[j + m + k] * eps[n / i * k];
buffer[j + m + k] = buffer[j + k] - z;
buffer[j + k] = buffer[j + k] + z;
}
}
}
ll n,k,block,num;
ll prec[30004],c[30040],suf[40000];
//bool vis[40000];
ll ans=0;
ll maxa=0;
struct Range{ ll l,r; } rg[1000];
struct Infor{ ll val,pos;} a[100005]; int main(){
read(n);
block=3;
if ((int)n/sqrt(n)>3) block=(int)n/sqrt(n);
FORP(i,1,n) {
read(a[i].val);
g[a[i].val].x++;
maxa=max(maxa,a[i].val);
a[i].pos=(i-1)/block+1;
suf[a[i].val]++;
} ll k=1; maxa*=2;
while (k<maxa) k<<=1;
k<<=1;
get_eps(k); num=a[n].pos;
FORP(i,1,num) rg[i].l=(i-1)*block+1,rg[i].r=min(n,(i)*block);
FORP(i,1,num){
FORP(j,rg[i].l,rg[i].r) c[a[j].val]++;
FORP(l,rg[i].l,rg[i].r-1){
suf[a[l].val]--;
FORP(r,l+1,rg[i].r) {
suf[a[r].val]--;
if (a[l].val==a[r].val) ans+=suf[a[l].val]+prec[a[l].val];
else
{
int x=a[l].val-(a[r].val-a[l].val);
if (x>0) ans+=prec[x];
x=a[r].val+(a[r].val-a[l].val);
if (x>0) ans+=suf[x];
}
}
FORP(r,l+1,rg[i].r) suf[a[r].val]++;
}
FORP(j,rg[i].l,rg[i].r) prec[a[j].val]+=c[a[j].val],c[a[j].val]=0;
suf[a[rg[i].r].val]--;
}
FORP(i,rg[1].l,rg[1].r) g[a[i].val].x--;
FORP(i,2,num-1){
//memset(f,0,sizeof(f)); memset(g,0,sizeof(g));
FORP(j,rg[i-1].l,rg[i-1].r) f[a[j].val].x++;
FORP(j,rg[i-1].l,rg[i-1].r) g[a[j].val].x--;
fft(k,f,eps); fft(k,g,eps);
FORP(j,0,k-1) f[j]=f[j]*g[j];
fft(k,f,inv_eps);
FORP(j,rg[i].l,rg[i].r) ans+=(ll)(f[a[j].val+a[j].val].x/k+0.5);
}
write(ans); puts("");
}
调下块大小a掉了
/*=================================
# Created time: 2016-04-19 16:09
# Filename: cccountari.cpp
# Description: Ploblem from codechef countari
=================================*/
#define me AcrossTheSky&HalfSummer11
#include <cstdio>
#include <cmath>
#include <ctime>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm> #include <set>
#include <stack>
#include <queue>
#include <vector> #define lowbit(x) (x)&(-x)
#define Abs(x) ((x) > 0 ? (x) : (-(x)))
#define FOR(i,a,b) for((i)=(a);(i)<=(b);(i)++)
#define FORP(i,a,b) for(int i=(a);i<=(b);i++)
#define FORM(i,a,b) for(int i=(a);i>=(b);i--)
#define ls(a,b) (((a)+(b)) << 1)
#define rs(a,b) (((a)+(b)) >> 1)
#define getlc(a) ch[(a)][0]
#define getrc(a) ch[(a)][1] #define maxn 100005
#define maxm 100005
#define INF 1070000000
using namespace std;
typedef long long ll;
typedef unsigned long long ull; template<class T> inline
void read(T& num){
num = 0; bool f = true;char ch = getchar();
while(ch < '0' || ch > '9') { if(ch == '-') f = false;ch = getchar();}
while(ch >= '0' && ch <= '9') {num = num * 10 + ch - '0';ch = getchar();}
num = f ? num: -num;
}
int outs[100];
template<class T> inline
void write(T x){
if (x==0) {putchar('0'); putchar(' '); return;}
if (x<0) {putchar('-'); x=-x;}
int num=0;
while (x){ outs[num++]=(x%10); x=x/10;}
FORM(i,num-1,0) putchar(outs[i]+'0'); //putchar(' ');
}
/*==================split line==================*/
const double pi=acos(-1);
struct cpx{
double x,y;
cpx(double a=0,double b=0):x(a),y(b){}
}f[maxn],g[maxn],eps[maxn],inv_eps[maxn]; inline cpx operator +(cpx a,cpx b){return cpx(a.x+b.x,a.y+b.y);}
inline cpx operator -(cpx a,cpx b){return cpx(a.x-b.x,a.y-b.y);}
inline cpx operator *(cpx a,cpx b){return cpx(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
inline cpx conj(cpx a){return cpx(a.x,-a.y);} inline void get_eps(int p){
double angle=2.0*pi/p;
FORP(i,0,p-1) eps[i]=cpx(cos(angle*i),sin(angle*i));
FORP(i,0,p-1) inv_eps[i]=conj(eps[i]);
}
inline void fft(int n,cpx* buffer,cpx* eps){
for (int i = 0, j = 0;i < n; i++){
if (i > j) swap(buffer[i],buffer[j]);
for (int l = n >> 1;(j ^= l) < l;l >>= 1);
}
for (int i = 2; i <= n; i <<= 1){
int m = i >> 1;
for (int j = 0; j < n; j += i)
for (int k = 0; k != m; k++){
cpx z = buffer[j + m + k] * eps[n / i * k];
buffer[j + m + k] = buffer[j + k] - z;
buffer[j + k] = buffer[j + k] + z;
}
}
}
int n,k,block,num;
int prec[30010],c[30010],suf[30100];
ll ans=0;
int maxa=0;
struct Range{ int l,r; } rg[10000];
struct Infor{ int val,pos;} a[maxn]; int main(){
read(n);
block=4000;
//if ((ll)n/sqrt(n)>3) block=(ll)n/sqrt(n);
block=min(block,n);
FORP(i,1,n) {
read(a[i].val);
maxa=max(maxa,a[i].val);
a[i].pos=(i-1)/block+1;
suf[a[i].val]++;
} int k=1; //maxa*=2;
while (k<maxa) k<<=1;
k<<=1;
get_eps(k); num=a[n].pos;
FORP(i,1,num) rg[i].l=(i-1)*block+1,rg[i].r=(i)*block; rg[num].r=n;
FORP(i,1,num){
FORP(j,rg[i].l,rg[i].r) c[a[j].val]++;
FORP(l,rg[i].l,rg[i].r-1){
suf[a[l].val]--;
FORP(r,l+1,rg[i].r) {
suf[a[r].val]--;
if (a[l].val==a[r].val) ans+=suf[a[l].val]+prec[a[l].val];
else
{
int x=a[l].val-(a[r].val-a[l].val);
if (x>0 && x<=maxa) ans+=prec[x];
x=a[r].val+(a[r].val-a[l].val);
if (x>0 && x<=maxa) ans+=suf[x];
}
}
FORP(r,l+1,rg[i].r) suf[a[r].val]++;
}
suf[a[rg[i].r].val]--;
FORP(j,0,maxa) f[j]=cpx(prec[j],0),g[j]=cpx(suf[j],0);
FORP(j,maxa+1,k) f[j]=cpx(0.0,0),g[j]=cpx(0.0,0);
fft(k,f,eps); fft(k,g,eps);
FORP(j,0,k-1) f[j]=f[j]*g[j];
fft(k,f,inv_eps);
FORP(j,rg[i].l,rg[i].r) ans+=trunc(f[a[j].val+a[j].val].x/k+0.5); FORP(j,rg[i].l,rg[i].r) prec[a[j].val]+=c[a[j].val],c[a[j].val]=0; }
/*FORP(i,2,num-1){
memset(f,0,sizeof(f)); memset(g,0,sizeof(g));
FORP(j,1,rg[i].l-1) f[a[j].val].x++;
FORP(j,rg[i].r+1,n) g[a[j].val].x++;
fft(k,f,eps); fft(k,g,eps);
FORP(j,0,k-1) f[j]=f[j]*g[j];
fft(k,f,inv_eps);
//FORP(j,0,maxa) printf("%d ",(int)(f[j].x/k+0.5)); puts("");
FORP(j,rg[i].l,rg[i].r) ans+=(int)(f[a[j].val+a[j].val].x/k+0.5);
}*/
write(ans); puts("");
//printf("%lf",clock()-t);
}
CC countari & 分块+FFT的更多相关文章
- BZOJ 3509 [CodeChef] COUNTARI ——分块 FFT
分块大法好. 块内暴力,块外FFT. 弃疗了,抄SX队长$silvernebula$的代码 #include <map> #include <cmath> #include & ...
- BZOJ 3509 分块FFT
思路: 跟今年WC的题几乎一样 (但是这道题有重 不能用bitset水过去) 正解:分块FFT http://blog.csdn.net/geotcbrl/article/details/506364 ...
- CC Arithmetic Progressions (FFT + 分块处理)
转载请注明出处,谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents by---cxlove 题目:给出n个数,选出三个数,按下标顺序形成等差数 ...
- CodeChef COUNTARI Arithmetic Progressions（分块 + FFT）
题目 Source http://vjudge.net/problem/142058 Description Given N integers A1, A2, …. AN, Dexter wants ...
- BZOJ3509 [CodeChef] COUNTARI 【分块 + fft】
题目链接 BZOJ3509 题解 化一下式子,就是 \[2A[j] = A[i] + A[k]\] 所以我们对一个位置两边的数构成的生成函数相乘即可 但是由于这样做是\(O(n^2logn)\)的,我 ...
- bzoj 3509: [CodeChef] COUNTARI] [分块 生成函数]
3509: [CodeChef] COUNTARI 题意:统计满足\(i<j<k, 2*a[j] = a[i] + a[k]\)的个数 \(2*a[j]\)不太好处理,暴力fft不如直接暴 ...
- 【做题】codechefCOUNTARI——分块FFT
记本题数组长度为\(n\),权值大小为\(m\). 首先,暴力显然是\(O(n^2)\)的. 先瞄一眼tag,然后发现这是FFT. 显然,问题的关键在于要满足i,j,k之间的位置关系.于是考虑分治FF ...
- cf 472G Design Tutorial: Increase the Constraints 分块+压位/FFT
题目大意 给出两个\(01\)序列\(A\)和\(B\) 哈明距离定义为两个长度相同的序列中,有多少个对应位置上的数字不一样 "00111" 和 "10101" ...
- luogu1919 A*BProblem升级版 (FFT)
把一个n位数看做n-1次的多项式,每一项的系数是反过来的每一位最后每一项系数进进位搞一搞就行了(数组一定要开到2的次数..要不然极端数据会RE) #include<cstdio> #inc ...
随机推荐
- 天河微信小程序入门《三》：打通任督二脉，前后台互通
原文链接:http://www.wxapp-union.com/forum.php?mod=viewthread&tid=505&extra=page%3D1 天河君在申请到https ...
- WebForm 分页、组合查询--2017年1月5日
sql = "select * from Commodity"; hs = new Hashtable(); if (txt_name.Text.Trim() != "& ...
- C#发送邮件异常：根据验证过程,远程证书无效
今天在做发送邮件功能时,开始用qq邮箱和163邮箱都可以正常发送,后再改用我公司的邮箱和smtp时竟然报错了. 异常提示-----"根据验证过程,远程证书无效",后来通过查询资料解 ...
- Builder模式在Java中的应用
在设计模式中对Builder模式的定义是用于构建复杂对象的一种模式,所构建的对象往往需要多步初始化或赋值才能完成.那么,在实际的开发过程中,我们哪些地方适合用到Builder模式呢?其中使用Build ...
- Angular2 组件生命周期
1. 说明 Angular每个组件都存在一个生命周期,从创建,变更到销毁.Angular提供组件生命周期钩子,把这些关键时刻暴露出来,赋予在这些关键结点和组件进行交互的能力. 2. 接口 按照生命周期 ...
- c#反射机制
一:反射的定义 审查元数据并收集关于它的类型信息的能力.元数据(编译以后的最基本数据单元)就是一大堆的表,当编译程序集或者模块时,编译器会创建一个类定义表,一个字段定义表,和一个方法定义表等. Sys ...
- WinForm 遍历用户控件里CheckBox
1.常用调用方法 public partial class UCRights : UserControl { private readonly int LOCATIONY; private DataT ...
- Yii 1开发日记 -- 搜索功能及Checkbox的实现
用yii 1实现后台的搜索功能,效果如下图: 1.模型中: public function search() { $criteria = new CDbCriteria; //独立高级搜索 if(is ...
- quicktest Professional下载地址，无限制使用方法
QTP是quicktest Professional的简称,是一种自动测试工具.使用QTP的目的是想用它来执行重复的自动化测试,主要是用于回归测试和测试同一软件的新版本.因此你在测试前要考虑好如何对应 ...
- 关于mysql MYISAM引擎的锁问题
CREATE TABLE `citymy` ( `city_id` smallint(5) unsigned NOT NULL DEFAULT '0', `city` varchar(50) CHAR ...