AtCoder Beginner Contest 258 A~Ex 题解

D - Trophy

题目大意

有一个游戏，由$N$个关卡组成。第$i$个关卡由一个数对$(A_i,B_i)$组成。

要通过一个关卡，你必须先花$A_i$的时间看一次介绍。然后，用$B_i$的时间打通这个关卡。若想多次通过同一个关卡，则第一次需要看介绍，后面无需再看（即如果想打通第$i$关$N$次，则所需时间为$A_i+N\times B_i$）。

开始时，只有第一关被解锁。当你每打通一关，其下一关会自动解锁（最后一关除外）。求总共打通$X$关的最少时间（重复通关也算）。

$1\le N\le 2\times 10^5$
$1\le A_i,B_i\le 10^9$
$1\le X\le N$

输入格式

$NX$
$A_1{B}_1$
$⋮$
$A_N{B}_N$

输出格式

输出答案。

样例

略，请自行前往AtCoder查看。

分析

仔细想想会发现，通过的方式都是先不重复通过某一关前所有关卡，再通过这一关一定次数，最终达到正好$X$次。因此，我们利用前缀和，依次枚举每个关卡，最终时间复杂度可达$\mathcal{O}(n)$。

代码

#include <cstdio>
#define INF 0x7FFFFFFFFFFFFFFFLL
using namespace std;

int main()
{
	int n, x;
	scanf("%d%d", &n, &x);
	if(x < n) n = x;
	long long s = 0LL, ans = INF, cur;
	for(int i=1; i<=n; i++)
	{
		int a, b;
		scanf("%d%d", &a, &b);
		if((cur = (s += a + b) + (long long)(x - i) * b) < ans)
			ans = cur;
	}
	printf("%lld\n", ans);
	return 0;
}

---

E - Packing Potatoes

题目大意

$10^{100}$个土豆排成一列。它们的重量分别是$W_0,W_1,\dots ,W_{N-1},W_0,\dots$，即第$i$个土豆的重量是$W_{i\mathrm{mod}N}$（$i=0,1,2,\dots$）。

Takahashi会往一个盒子里依次放入每个土豆，当放入的土豆总重量$\ge X$的时候他会换一个新盒子。

给定$Q$个询问。第$i$个询问：给定整数$K_i$，求第$K_i$个盒子中土豆的个数。

$1\le N,Q\le 2\times 10^5$
$1\le X,W_i\le 10^9$
$1\le K_i\le 10^{12}$

输入格式

$NQX$
$W_0\dots W_{N-1}$
$K_1$
$⋮$
$K_Q$

输出格式

输出$Q$行，第$i$行是第$i$个询问的答案。

样例

略，请自行前往AtCoder查看。

分析

周期问题。

代码

#include <cstdio>
#include <vector>
#define maxn 200005
using namespace std;

using LL = long long;
int cnt[maxn], ord[maxn], w[maxn << 1];

template <typename T>
inline T read()
{
	char c;
	while((c = getchar()) < '0' || c > '9');
	T res = c ^ 48;
	while((c = getchar()) >= '0' && c <= '9')
		res = (res << T(3)) + (res << T(1)) + (c ^ 48);
	return res;
}

inline void print(int x)
{
	if(x > 9) print(x / 10);
	putchar(x % 10 ^ 48);
}

inline void println(int x)
{
	print(x);
	putchar('\n');
}

int main()
{
	int n = read<int>(), q = read<int>(), x = read<int>();
	LL sum = 0LL;
	for(int i=0; i<n; i++)
		sum += w[i + n] = w[i] = read<int>();
	int fill = x / sum * n;
	for(int i=0; i<n; i++)
		cnt[i] = fill, ord[i] = -1;
	x %= sum;
	for(int i=0, j=0, s=0; i<n; i++)
	{
		if(j < i) j = i, s = 0;
		while(s < x)
		{
			s += w[j];
			j += 1;
		}
		cnt[i] += j - i;
		s -= w[i];
	}
	vector<int> path;
	int loop = -1;
	for(int u=0, k=0; ; k++)
	{
		if(ord[u] != -1)
		{
			loop = k - ord[u];
			break;
		}
		ord[u] = k;
		path.push_back(u);
		(u += cnt[u]) %= n;
	}
	int non_loop = path.size() - loop;
	while(q--)
	{
		LL k = read<LL>();
		println(cnt[path[--k < non_loop? k:
				non_loop + (k - non_loop) % loop]]);
	}
	return 0;
}

---

F - Main Street

WJ...

---

G - Triangle

题目大意

给定一张由$N$个点组成的简单无向图$G$和它的邻接矩阵$A$。

什么是邻接矩阵 ？

• 邻接矩阵，顾名思义，即表示图中每两点之间关系的矩阵。
• 如本题中$A_{i,j}$表示$i,j$两点之间是否有边。$A_{i,j}=0$表示无边，$1$表示有边。
• 一般来说，对于任意简单无向图，$A_{i,i}=0$，$A_{i,j}=A_{j,i}$ ($i\ne j$)。

求数对$(i,j,k)$的个数，使得：

• $1\le i<j<k\le N$
• $(i,j,k)$三个点在图中组成一个三角形，即$i,j,k$中任意两点之间有一条连边。

$3\le N\le 3000$

输入格式

$N$
$A_{1,1}{A}_{1,2}\dots A_{1,N}$
$A_{2,1}{A}_{2,2}\dots A_{2,N}$
$⋮$
$A_{N,1}{A}_{N,2}\dots A_{N,N}$（注意没有空格，如10110）

输出格式

输出一个整数，即符合条件的数对$(i,j,k)$的个数。

样例

略，请自行前往AtCoder查看。

分析

前言

• 个人感觉这题其实是这场比赛中最有意思的题。题目不难，但很具有研究意义。
  这里我将从各个角度分析题目，也期待大家在评论区中分享其他方法。

废话不多说，题解开始！

题目说得太啰嗦，其实不用管什么图论，可以看成是给定$N\times N$的$01$矩阵$A$，求$A_{i,j}=A_{i,k}=A_{j,k}=1$的$(i,j,k)$的个数。

再来看数据范围。$N\le 3000$，则$\mathcal{O}(N^3)$的朴素算法时间可达到$T(2.7\times 10^{10})$，显然无法通过。

然而，事实并非如此。
在仔细研究后发现，时间限制为$3\mathrm{s}$的题目可以使用复杂度稍高的算法。
但是一般来说，极端情况下超过$T(10^{10})$的算法无法通过。
因此，也许是数据不够强吧，使用如下的优化可以恰好通过（#32949139 $37764\mathrm{KB},2755\mathrm{ms}$）：

#pragma GCC optimize("Ofast") // -Ofast 预编译优化
#include <cstdio>
#define maxn 3000 // 数组大小设置正好，减少内存占用
using namespace std;

// 题目中正常的邻接矩阵
bool a[maxn][maxn];

// 特殊邻接表存储，减少尝试次数
// 这里不使用vector，会拖慢速度
int G[maxn][maxn], sz[maxn];

inline void print(const long long& x) // 快写-输出优化
{
	if(x > 9LL) print(x / 10);
	putchar(x % 10 ^ 48);
}

int main()
{
	// getchar快读输入优化
	int n = 0; char c;
	while((c = getchar()) != '\n')
		n = (n << 3) + (n << 1) + (c ^ 48);
	for(int i=0; i<n; i++, getchar())
		for(int j=0; j<n; j++)
			if(getchar() == '1' && j > i) // j > i：只存一半，去掉重复
				a[i][j] = 1, G[i][sz[i]++] = j;

	// 注意答案数据类型使用long long
	long long ans = 0LL;
	for(int v=0; v<n; ++v)
		for(int i=0; i<sz[v]; ++i) // 直接调取邻接表，省去不必要判断
		{
			int u = G[v][i];
			for(int j=0; j<sz[u]; ++j)
				if(a[v][G[u][j]])
					ans ++;
		}
	print(ans);
	return 0;
}

当然，这种方法并非每道题都能用，因此还是建议大家继续看正解。

正解还是基于上面的朴素算法，可看作：

• 依次遍历$A_{i,j}=1$的$(i,j)$（$i<j$）
• => 将答案加上$A[i]$和$A[j]$对应位置上都是$1$的位置个数
• 输出答案，除以3（去掉每个重复算的三次）

那么别的地方都没办法，只有第二步可以使用bitset的并集(&)操作进行优化。此时时间复杂度为$\mathcal{O}(\frac{n^3}{w})$，其中$w=64$。详见代码。

代码

#include <cstdio>
#include <bitset>
#define maxn 3000
using namespace std;

bitset<maxn> a[maxn];

int main()
{
	int n = 0; char c;
	while((c = getchar()) != '\n')
		n = (n << 3) + (n << 1) + (c ^ 48);
	for(int i=0; i<n; i++, getchar())
		for(int j=0; j<n; j++)
			if(getchar() == '1')
				a[i].set(j); // a[i][j] = 1
	long long ans = 0LL;
	for(int i=0; i+1<n; i++)
		for(int j=i+1; j<n; j++)
			if(a[i][j])
				ans += (a[i] & a[j]).count(); // 取交集，数1的个数
	printf("%lld\n", ans / 3LL);
	return 0;
}