Skip to content

Instantly share code, notes, and snippets.

@vegito2002
vegito2002 / pytorch_mnist.py
Created April 13, 2018 04:01 — forked from xmfbit/pytorch_mnist.py
an example of pytorch on mnist dataset
import torch
import torch.nn as nn
from torch.autograd import Variable
import torchvision.datasets as dset
import torchvision.transforms as transforms
import torch.nn.functional as F
import torch.optim as optim
## load mnist dataset
use_cuda = torch.cuda.is_available()
@vegito2002
vegito2002 / Solution.java
Last active March 27, 2018 13:17
frog v2ex
/*
V2不能改内容所以直接这里改了: maxqueue的做法实际上是错误的; 这题并不需要这么复杂;
直接把dp算出来, 然后最后单独走一个min_step的循环, 找一个最大值就行了;
在算每一个dp[i]的时候, 不能包含这个反向min_step的循环的最大值. 只能最后返回之前算这个;
*/
class Solution {
int solve (int[] staircase, int[] possible_steps) {
int min_step = Integer.MAX_VALUE, N = staircase.length;
for (int step : possible_steps)
@vegito2002
vegito2002 / V2EXJumpingFrog.java
Last active March 27, 2018 05:42
you are a frog
class Solution {
int solve (int[] staircase, int[] possible_steps) {
int min_step = Integer.MAX_VALUE, N = staircase.length;
for (int step : possible_steps)
min_step = Math.min (min_step, step);
if (min_step > N)
return 0;
int[] dp = new int[N];
for (int i = -1 + min_step; i < N; i++) {
int cur = Integer.MIN_VALUE;
@vegito2002
vegito2002 / Script.c
Created December 30, 2017 04:15
Non-decreasing sequence generation
#include <stdio.h>
#include <stdlib.h>
static int N;
static int counter = 0;
static int do_io = 0;
static void helper (int idx, int pos, int* triplet);
int