【AtCoder】AGC023 A-F题解

可以说是第一场AGC了，做了三道题之后还有30min，杠了一下D题发现杠不出来，三题滚粗了

rating起步1300+，感觉还是很菜。。。

只有三题水平显然以后还会疯狂--啊（CF的惨痛经历）

改题的感觉似乎还不错因为思维都非常的妙（我根本想不到）

A - Zero-Sum Ranges

开场娱乐大家的小水题，区间和为0的情况存在于sum[R] == sum[L - 1]，只要记录一下某一个值的sum出现了多少次就行，懒得离散化直接用map就OK啊

代码

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

#include <map>

#define MAXN 200005

#define PLI pair<long long,int>

#define fi first

#define se second

#define mp make_pair

//#define ivorysi

using namespace std;

typedef long long int64;

int N;

int64 A[MAXN];

map<int64,int> mmm;

void Solve() {

	scanf("%d",&N);

	for(int i = 1 ; i <= N ; ++i) scanf("%lld",&A[i]);

	mmm[0] = 1;

	int64 ans = 0;

	for(int i = 1 ; i <= N ; ++i) {

		A[i] += A[i - 1];

		ans += mmm[A[i]];

		mmm[A[i]] += 1;

	}

	printf("%lld\n",ans);

}

int main() {

#ifdef ivorysi

	freopen("f1.in","r",stdin);

#endif

	Solve();

	return 0;

}

B - Find Symmetries

把矩阵行循环平移x，列循环平移y后，矩阵关于左上到右下这条对角线对称

矩阵大小是300

然后我就写了个暴力然而我并没发现我的暴力是\(n^4\)然后愉快TLE

我就开始想着优化，发现固定行平移，每平移一列hash值的更改可以O1算出来，check把每行每列hash起来比较，是O(n)的，然后过掉了

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

#include <map>

#define MAXN 200005

#define PLI pair<long long,int>

#define fi first

#define se second

#define mp make_pair

#define ha 99994711

#define ba 823

//#define ivorysi

using namespace std;

typedef long long int64;

int N;

char s[305][305],b[305][305];

int64 hc[305],hr[305],e;

inline int C(int x) {

	return x > N ? x - N : x;

}

bool check(int y) {

	int T = C(N + y);

	for(int i = 1 ; i <= N ; ++i)

		hr[i] = (hr[i] * ba + (b[i][T] - 'a' + 1)) % ha;

	for(int i = 1 ; i <= N ; ++i) {

		if(hr[i] != hc[C(i + y)]) {

			T = C(N + y + 1);

			for(int j = 1 ; j <= N ; ++j)

				hr[j] = (hr[j] - e * (b[j][T] - 'a' + 1) % ha + ha) % ha;

			return false;

		}

	}

	T = C(N + y + 1);

	for(int i = 1 ; i <= N ; ++i)

		hr[i] = (hr[i] - e * (b[i][T] - 'a' + 1) % ha + ha) % ha;

	return true;

}

void Solve() {

	scanf("%d",&N);

	if(N == 1) {

		puts("1");return;

	}

	e = 1;

	for(int i = 1 ; i < N ; ++i) e = e * ba % ha;

	for(int i = 1 ; i <= N ; ++i) {

		scanf("%s",s[i] + 1);

	}

	int cnt = 0;

	for(int A = 0 ; A < N ; ++A) {

		memset(hc,0,sizeof(hc));

		memset(hr,0,sizeof(hr));

		for(int i = 1 ; i <= N ; ++i) {

			for(int j = 1 ; j <= N ; ++j) {

				b[i][j] = s[C(i + A)][j];

				hc[j] = (hc[j] * ba + (b[i][j] - 'a' + 1)) % ha;

				if(j != N) {

					hr[i] = (hr[i] * ba + (b[i][j] - 'a' + 1)) % ha;

				}

			}

		}

		for(int B = 0 ; B < N ; ++B) {

			if(check(B)) ++cnt;

		}

	}

	printf("%d\n",cnt);

}

int main() {

#ifdef ivorysi

	freopen("f1.in","r",stdin);

#endif

	Solve();

	return 0;

}

C - Painting Machines

while(1) 推式子

联想到地震后的幻想乡显然题目可以转化为

\(\sum_{i = 0}^{N - 2} T(i)\)其中\(T(i)\)表示用了i次操作没有全部染黑的方案数

有\(T(0) = (N - 1)!\)

然后分类讨论，一个是因为没用N-1而没有全部染黑的方案数是\(\binom{N - 2}{i}i!\)

如果用了N - 1，那么方案是\((\binom{N - 2}{i - 1} - W(i - 1))i!\)

\(W(i)\)表示用了一个N - 1号机器和其他i个机器，把所有方格染黑了的方案数

显然序列里会有1

把序列差分一下，会发现这个序列不是1就是2，也就是一堆1和一堆2的数目固定，一堆1和一堆2的和是N - 2，这是个二元一次方程，然后求出了1的个数和2的个数，就是个带重复元素的全排列，答案是

\(\frac{(num_1 + num_2)! }{num_1 ! num_2 !}\)

其他\(N - 1 - i\)台机器随意排列，最后还要乘上\((N - i - 1)!\)

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

#include <map>

#define MAXN 1000005

#define PLI pair<long long,int>

#define fi first

#define se second

#define mp make_pair

#define ha 99994711

#define ba 823

#define MOD 1000000007

//#define ivorysi

using namespace std;

typedef long long int64;

int64 fac[MAXN],invfac[MAXN],ans;

int N;

int64 fpow(int64 x,int64 c) {

	int64 res = 1,t = x;

	while(c) {

		if(c & 1) res = res * t % MOD;

		t = t * t % MOD;

		c >>= 1;

	}

	return res;

}

int64 C(int n,int m) {

	if(n < m) return 0;

	return fac[n] * invfac[m] % MOD * invfac[n - m] % MOD;

}

int64 W(int k) {

	int x = N - 2 - k;

	int y = k - x;

	if(x < 0 || y < 0) return 0;

	return fac[k] * invfac[y] % MOD * invfac[x] % MOD;

}

void Solve() {

	scanf("%d",&N);

	if(N == 2) {

		puts("1");

		return;

	}

	fac[0] = invfac[0] = 1;

	for(int i = 1 ; i <= N ; ++i) {

		fac[i] = fac[i - 1] * i % MOD;

	}

	invfac[N] = fpow(fac[N],MOD - 2);

	for(int i = N - 1 ; i >= 1 ; --i) {

		invfac[i] = invfac[i + 1] * (i + 1) % MOD;

	}

	ans += fac[N - 1];

	for(int i = 1 ; i < N - 1 ; ++i) {

		ans += (C(N - 2,i) + C(N - 2,i - 1) + MOD - W(i - 1))* fac[i] % MOD * fac[N - 1 - i] % MOD;

		ans %= MOD;

	}

	printf("%lld\n",ans);

}

int main() {

#ifdef ivorysi

	freopen("f1.in","r",stdin);

#endif

	Solve();

	return 0;

}

D - Go Home

题解

神仙博弈题（其实也不算博弈）

你看完题，发现，啥，最优决策，咋最优啊，啥最优啊，什么玩意，弃疗吧……

题解是这么说的，如果汽车在1 和N中间，且\(A_1 >= A_n\)，那么N肯定会投票给1，为什么，因为赢不了，如果暂时让车靠近了N，最后也会因为N - 1公寓里的人都回家了而车也再回到1，所以为了早回家，N的票都会给1，且最后的操作一定会有一个\(X_{n} - X_{1}\)的长度，那么我们把N的票全部给1，\(P_{1} += P_{N}\)也没有问题，如果\(A_{n} > A_{1}\)是类似的，之后就变成了1和N - 1之间的子问题……停止条件是所有公寓都在初始位置\(S\)的一边

累加答案的话要注意和前一次操作方向相反才累加，如果前一次1 - N，下一次1 - (N - 1)，第二次的距离不用累加进答案

代码

#include <iostream>

#include <cstdio>

#define MAXN 100005

//#define ivorysi

using namespace std;

typedef long long int64;

typedef double db;

int N;

int64 S,X[MAXN],P[MAXN],ans;

void Solve() {

    scanf("%d%lld",&N,&S);

    for(int i = 1 ; i <= N ; ++i) {

    	scanf("%lld%lld",&X[i],&P[i]);

    }

    int L = 1,R = N;

    int dir = 0;

    while(1) {

    	if(X[L] >= S) {ans += X[R] - S;break;}

    	if(X[R] <= S) {ans += S - X[L];break;}

    	if(P[L] >= P[R]) {

    		if(dir != 1) {dir = 1;ans += X[R] - X[L];}

    		P[L] += P[R];R--;

    	}

    	else {

    		if(dir != 2) {dir = 2;ans += X[R] - X[L];}

    		P[R] += P[L];L++;

    	}

    }

    printf("%lld\n",ans);

}

int main() {

#ifdef ivorysi

    freopen("f1.in","r",stdin);

#endif

    Solve();

    return 0;

}

E - Inversions

我觉得这道题计算排列的方法应该算个比较重要前置技能，但这又不是什么板啥的，如果知道了这个子问题那么这道题至少会有头绪

如何计算每个数位有限制的排列总个数？

先计算一个\(cnt[k]\)表示可以填k的位置的个数，可以用一个后缀和算出来

答案就是\(\prod_{i = 1}^{N} cnt[i] - (N - i)\)

有了这个我们再来看这道题

我们对于两个位置\(i < j\)且\(A_{i} <= A_{j}\)计算\(i\)位置上的数大于\(j\)位置上的数的排列的总和，我们可以让\(A_{j} = A_{i}\)，然后计算排列个数，再除二就是答案

以下的\(cnt[i]\)更改成\(cnt[i] - (N - i)\)

这样的话我们考虑更改\(A_{j}\)会使得\([A_{i} + 1,A_{j}]\)这个区间的cnt减1，也就是乘上了\((cnt[i] - 1) / cnt[i]\)，也就是处理成区间前缀乘积后相除，然而会发现处理0的情况比较特殊，还需要记录前缀出现了几个0，前缀上0个数相同值才不为0，位置增加0的个数单调不减，所以可以预处理出来每一段开始位置和结束位置

\(A_{i} > A_{j}\)我们发现就是把\(A_{i} := A_{j}\)后总排列数减去计算出的新排列值的一半

以上两种情况都可以用树状数组快速维护

代码

#include <iostream>

#include <cstdio>

#include <cstring>

#define MAXN 200005

//#define ivorysi

using namespace std;

typedef long long int64;

typedef double db;

const int MOD = 1000000007;

int N,A[MAXN],x[MAXN],st[MAXN],ed[MAXN];

int64 cnt[MAXN],V[MAXN],D[MAXN],ID[MAXN],S,tr[MAXN],tr_cnt[MAXN],Inv_2,ans;

int64 fpow(int64 x,int64 c) {

	int64 res = 1,t = x;

	while(c) {

		if(c & 1) res = res * t % MOD;

		t = t * t % MOD;

		c >>= 1;

	}

	return res;

}

int lowbit(int x) {return x & (-x);}

void insert(int x,int64 v) {

	while(x <= N) {

		tr[x] += v;

		tr[x] %= MOD;

		tr_cnt[x]++;

		x += lowbit(x);

	}

}

int64 Query(int x,int64 v) {

	int64 res = 0;

	while(x > 0) {

		res += tr[x];

		x -= lowbit(x);

	}

	res %= MOD;

	return res * v % MOD;

}

int64 getnum(int x) {

	int64 res = 0;

	while(x > 0) {

		res += tr_cnt[x];

		x -= lowbit(x);

	}

	return res;

}

int64 Query_range(int L,int R,int64 v) {

	if(L == 0) ++L;

	if(L > R) return 0;

	return (Query(R,v) + MOD - Query(L - 1,v)) % MOD;

}

void Solve() {

    scanf("%d",&N);

    for(int i = 1 ; i <= N ; ++i) scanf("%d",&A[i]),cnt[A[i]]++;

    for(int i = N - 1; i >= 1 ; --i) cnt[i] += cnt[i + 1];

    S = 1;Inv_2 = (MOD + 1) / 2;

    for(int i = 1 ; i <= N ; ++i) {

    	cnt[i] -= N - i;

    	if(cnt[i] <= 0) {puts("0");return;}

    	S = S * cnt[i] % MOD;

    }

    for(int i = 1 ; i <= N ; ++i) {

    	V[i] = (cnt[i] - 1) * fpow(cnt[i],MOD - 2) % MOD;

    }

    D[0] = 1;

    st[0] = 0;

    for(int i = 1 ; i <= N ; ++i) {

    	x[i] = x[i - 1];D[i] = D[i - 1];

    	if(V[i] == 0) {x[i]++;st[x[i]] = i;ed[x[i] - 1] = i - 1;}

    	else D[i] = D[i] * V[i] % MOD;

    	ID[i] = fpow(D[i],MOD - 2);

    }

    ed[x[N]] = N;

    for(int i = 1 ; i <= N ; ++i) {

    	ans += Query_range(st[x[A[i]]],A[i],D[A[i]] * S % MOD * Inv_2 % MOD);

    	ans %= MOD;

    	insert(A[i],ID[A[i]]);

    }

    memset(tr,0,sizeof(tr));

    memset(tr_cnt,0,sizeof(tr_cnt));

    for(int i = 1 ; i <= N ; ++i) {

    	int64 t = Query_range(A[i] + 1,ed[x[A[i]]],ID[A[i]] * S % MOD * Inv_2 % MOD);

    	ans += ((getnum(N) - getnum(A[i])) * S % MOD + MOD - t) % MOD;

    	ans %= MOD;

    	insert(A[i],D[A[i]]);

    }

    printf("%lld\n",ans);

}

int main() {

#ifdef ivorysi

    freopen("f1.in","r",stdin);

#endif

    Solve();

    return 0;

}

F - 01 on Tree

题解

这个是真的神仙题。。。

脑子里一直是某个按子树大小的贪心，然后一直被叉掉

最后说是初始化每个节点就是一个单独的树，记录这个树里1的个数和0，初始的时候1的个数即为节点值是否为1,0的个数即为节点值是否为0

我们按照\(C_{0i}/C_{1i}\)排序，并把每个点和直接父亲接起来，意思就是把这个点的0和1接在父亲节点的后面，多出来的贡献即是

v是儿子，p是父亲，\(C_{1p} * C_{0v}\)

推一下式子可以证出来选别的儿子肯定不优

然后更新父亲p的值，也就是p变成了一个新的树，这个可以用并查集维护

然后带更新的排序用一个set实现就可以

神仙题神仙题。。。

代码

#include <iostream>

#include <cstdio>

#include <vector>

#include <set>

#include <cstring>

#define MAXN 200005

//#define ivorysi

using namespace std;

typedef long long int64;

typedef double db;

int N,P[MAXN],V[MAXN],fa[MAXN],C[2][MAXN];

int getfa(int x) {

	return fa[x] == x ? x : fa[x] = getfa(fa[x]);

}

struct cmp {

	bool operator() (const int &a,const int &b) {

		if(1LL * C[0][a] * C[1][b] != 1LL * C[0][b] * C[1][a])

			return 1LL * C[0][a] * C[1][b] > 1LL * C[0][b] * C[1][a];

		else return a > b;

	}

};

multiset<int,cmp> S;

void Solve() {

	scanf("%d",&N);

    for(int i = 2 ; i <= N ; ++i) scanf("%d",&P[i]);

   	for(int i = 1 ; i <= N ; ++i) scanf("%d",&V[i]);

   	for(int i = 1 ; i <= N ; ++i) fa[i] = i;

   	for(int i = 1 ; i <= N ; ++i) C[V[i]][i] = 1;

   	for(int i = 2 ; i <= N ; ++i) S.insert(i);

   	int64 ans = 0;

   	for(int i = 1 ; i <= N - 1 ; ++i) {

   		int v = *S.begin();

   		int p = getfa(P[v]);

   		S.erase(v);S.erase(p);

   		ans += 1LL * C[0][v] * C[1][p];

   		C[1][p] += C[1][v];C[0][p] += C[0][v];

   		fa[v] = p;

   		if(p != 1) S.insert(p);

   	}

   	printf("%lld\n",ans);

}

int main() {

#ifdef ivorysi

    freopen("f1.in","r",stdin);

#endif

    Solve();

    return 0;

}

巴特西