E. Li Hua and Array

Problem - E - Codeforces

E. Li Hua and Array_第1张图片

思路:观察给定的函数,其实就是求与这个数互质的数的个数,即欧拉函数,我们发现一个数迭代欧拉函数不会很多,那么对于第一个操作来说我们可以直接暴力修改,而对于第二个操作来说,就是求l,r的最近公共祖先,那么我们可以用线段树维护区间的最近公共祖先,并且由于迭代的次数很少,所以并不需要建图,直接暴力跳跃求最近公共祖先即可,那么最总的答案就是用l到r的深度之和-最近公共祖先的深度,乘以区间长度,这就是把这每个点跳跃到最近公共祖先的花费

一定不要建图,建图会MLE

// Problem: E. Li Hua and Array
// Contest: Codeforces - Codeforces Round 864 (Div. 2)
// URL: https://codeforces.com/problemset/problem/1797/E
// Memory Limit: 512 MB
// Time Limit: 3000 ms

#include
#include
#include
#define fi first
#define se second
#define i128 __int128
using namespace std;
typedef long long ll;
typedef double db;
typedef pair PII;
const double eps=1e-7;
const int N=5e6+7 ,M=6e7+7, INF=0x3f3f3f3f,mod=1e9+7,mod1=998244353;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}

int T,hackT;
int n,m,k;
int cost[N];
int pr[3000],cnt;
bool st[3000];
int vis[N];
int depth[N];
struct Node{
	int l,r;
	int start;
	int sum;
	int add;
};
Node tr[N*4];

int lca(int a,int b) {
	while(a!=b) {
		if(a>b) a=vis[a];
		else b=vis[b];
	}
	return a;
}

void pushup(int u) {
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
	tr[u].add=tr[u<<1].add+tr[u<<1|1].add;
	tr[u].start=lca(tr[u<<1].start,tr[u<<1|1].start);
}

void build(int u,int l,int r) {
	if(l==r) {
		if(depth[cost[l]]==0) tr[u]={l,r,cost[l],depth[cost[l]],1};
		else tr[u]={l,r,cost[l],depth[cost[l]],0};
	}else {
		tr[u]={l,r,INF,0,0};
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		
		pushup(u);
	}
}

void modify(int u,int l,int r) {
	if(tr[u].r-tr[u].l+1==tr[u].add) return ;
	if(tr[u].l==tr[u].r) {
		int tp=vis[tr[u].start];
		if(depth[tp]==0) tr[u]={tr[u].l,tr[u].r,tp,depth[tp],1};
		else tr[u]={tr[u].l,tr[u].r,tp,depth[tp],0}; 
	}else {
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1,l,r);
		if(r>mid) modify(u<<1|1,l,r);
		
		pushup(u);
	}
}

int query_sum(int u,int l,int r) {
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
	else {
		int mid=tr[u].l+tr[u].r>>1;
		int res=0;
		if(l<=mid) res+=query_sum(u<<1,l,r);
		if(r>mid) res+=query_sum(u<<1|1,l,r);
		
		return res;
	}
}

int query_lca(int u,int l,int r) {
	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].start;
	else {
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid&&r>mid) {
			int a=query_lca(u<<1,l,r);
			int b=query_lca(u<<1|1,l,r);
			
			return lca(a,b);
		}else if(l<=mid) return query_lca(u<<1,l,r);
		else return query_lca(u<<1|1,l,r);
	}
}

void get(int x) {
	int tx=x;
	int res=x;
	for(int i=0;i

你可能感兴趣的:(codeforces,算法)